diff --git a/src/sempy_labs/directlake/_update_directlake_partition_entity.py b/src/sempy_labs/directlake/_update_directlake_partition_entity.py
index d06c76e4..6822a69e 100644
--- a/src/sempy_labs/directlake/_update_directlake_partition_entity.py
+++ b/src/sempy_labs/directlake/_update_directlake_partition_entity.py
@@ -71,7 +71,7 @@ def update_direct_lake_partition_entity(
                     f"{icons.red_dot} The '{tName}' table in the '{dataset}' semantic model has not been updated."
                 )
 
-            tom.model.Tables[tName].Partitions[part_name].EntityName = eName
+            tom.model.Tables[tName].Partitions[part_name].Source.EntityName = eName
             print(
                 f"{icons.green_dot} The '{tName}' table in the '{dataset}' semantic model has been updated to point to the '{eName}' table."
             )