diff --git a/backend/python/pydevlake/pydevlake/domain_layer/code.py b/backend/python/pydevlake/pydevlake/domain_layer/code.py
index cffe7f50297..bd28802c907 100644
--- a/backend/python/pydevlake/pydevlake/domain_layer/code.py
+++ b/backend/python/pydevlake/pydevlake/domain_layer/code.py
@@ -50,7 +50,7 @@ class PullRequest(DomainModel, table=True):
 class PullRequestLabels(NoPKModel, table=True):
     __tablename__ = 'pull_request_labels'
     pull_request_id: str = Field(primary_key=True)
-    label_name: str
+    label_name: str = Field(primary_key=True)
 
 
 class PullRequestCommit(NoPKModel, table=True):
@@ -94,7 +94,7 @@ class Commit(NoPKModel, table=True):
 class CommitParent(NoPKModel, table=True):
     __tablename__ = 'commit_parents'
     commit_sha: str = Field(primary_key=True)
-    parent_commit_sha: str
+    parent_commit_sha: str = Field(primary_key=True)
 
 
 class CommitsDiff(DomainModel, table=True):
@@ -112,6 +112,7 @@ class RefCommit(NoPKModel, table=True):
     new_commit_sha: str
     old_commit_sha: str
 
+
 class Component(NoPKModel, table=True):
     __tablename__ = 'components'
     repo_id: str