diff --git a/src/sagemaker/estimator.py b/src/sagemaker/estimator.py
index fa40719c9f..df3f3cab02 100644
--- a/src/sagemaker/estimator.py
+++ b/src/sagemaker/estimator.py
@@ -1711,6 +1711,7 @@ def deploy(
             model = self._compiled_models[family]
         else:
             kwargs["model_kms_key"] = self.output_kms_key
+            kwargs["code_location"] = self.code_location
             model = self.create_model(**kwargs)
 
         model.name = model_name