diff --git a/src/based/columns.py b/src/based/columns.py
index 13a5267..da35c87 100644
--- a/src/based/columns.py
+++ b/src/based/columns.py
@@ -13,7 +13,6 @@ class ColumnDefinition:
         raise NotImplementedError
 
 
-# TODO: Not so good implementation (maybe)
 class UniqueColumnDefinition(ColumnDefinition):
     def __init__(self, wrapped: ColumnDefinition):
         super().__init__(wrapped.name)
@@ -23,7 +22,7 @@ class UniqueColumnDefinition(ColumnDefinition):
         return SQL("{} UNIQUE").format(self.wrapped.sql())
 
     def serialize(self):
-        return self.wrapped.serialize()
+        return f"{self.wrapped.serialize()}:unique"
 
 
 def make_column_unique(column) -> UniqueColumnDefinition:
@@ -38,7 +37,7 @@ class PrimarySerialColumnDefinition(ColumnDefinition):
         return SQL("{} SERIAL PRIMARY KEY").format(Identifier(self.name))
 
     def serialize(self):
-        return f"{self.name}:int"
+        return f"{self.name}:serial:primary"
 
 
 class PrimaryUUIDColumnDefinition(ColumnDefinition):
@@ -51,7 +50,7 @@ class PrimaryUUIDColumnDefinition(ColumnDefinition):
         )
 
     def serialize(self):
-        return f"{self.name}:uuid"
+        return f"{self.name}:uuid:primary"
 
 
 class TextColumnDefinition(ColumnDefinition):