diff --git a/metadata-ingestion/src/datahub/ingestion/source/aws/sagemaker_processors/models.py b/metadata-ingestion/src/datahub/ingestion/source/aws/sagemaker_processors/models.py
index e82cfc58f75a74..eef2b26ee08f2e 100644
--- a/metadata-ingestion/src/datahub/ingestion/source/aws/sagemaker_processors/models.py
+++ b/metadata-ingestion/src/datahub/ingestion/source/aws/sagemaker_processors/models.py
@@ -418,7 +418,7 @@ def get_model_wu(
 
         model_image_groups: Set[str] = set()
         if model_image is not None:
-            model_uri_groups = self.lineage.model_image_to_groups.get(
+            model_image_groups = self.lineage.model_image_to_groups.get(
                 model_image, set()
             )