diff --git a/airflow/migrations/versions/0026_2_10_0_dag_schedule_dataset_alias_reference.py b/airflow/migrations/versions/0026_2_10_0_dag_schedule_dataset_alias_reference.py
index d266340312fe..f4c11a7b006a 100644
--- a/airflow/migrations/versions/0026_2_10_0_dag_schedule_dataset_alias_reference.py
+++ b/airflow/migrations/versions/0026_2_10_0_dag_schedule_dataset_alias_reference.py
@@ -45,7 +45,7 @@ def upgrade():
     """Add dag_schedule_dataset_alias_reference table."""
     op.create_table(
         "dag_schedule_dataset_alias_reference",
-        sa.Column("alias_id", sa.Integer(), nullable=False),
+        sa.Column("alias_id", sa.Integer(), primary_key=True, nullable=False),
         sa.Column("dag_id", StringID(), primary_key=True, nullable=False),
         sa.Column("created_at", airflow.utils.sqlalchemy.UtcDateTime(timezone=True), nullable=False),
         sa.Column("updated_at", airflow.utils.sqlalchemy.UtcDateTime(timezone=True), nullable=False),
diff --git a/docs/apache-airflow/img/airflow_erd.sha256 b/docs/apache-airflow/img/airflow_erd.sha256
index c4958d617826..e8ee39a6eba2 100644
--- a/docs/apache-airflow/img/airflow_erd.sha256
+++ b/docs/apache-airflow/img/airflow_erd.sha256
@@ -1 +1 @@
-5067b6eac290220d23ce1bef6ce6d02478757f626f5a81a53ca2bba8e31c094c
\ No newline at end of file
+5958e108ee31f4ca94d6a878767d8d1ca00ef4e664ecda9308b367c173a09226
\ No newline at end of file