diff --git a/pkg/models/table.go b/pkg/models/table.go
index f6f4c7d..0695641 100644
--- a/pkg/models/table.go
+++ b/pkg/models/table.go
@@ -61,6 +61,7 @@ func (t *Table) setPrimaryKey(con *ast.Constraint) {
 		t.PrimaryKey.Columns[i].InUniqueKey = true
 	}
 
+	// TODO: remove this once the parser is fixed
 	for _, col := range t.Columns {
 		for _, pk := range t.PrimaryKey.Columns {
 			if col.Name == pk.Name {