diff --git a/sagemaker-train/tests/unit/train/test_base_trainer_serverful.py b/sagemaker-train/tests/unit/train/test_base_trainer_serverful.py
index 18cc798475..590a5cc987 100644
--- a/sagemaker-train/tests/unit/train/test_base_trainer_serverful.py
+++ b/sagemaker-train/tests/unit/train/test_base_trainer_serverful.py
@@ -31,6 +31,7 @@ def __init__(self, **kwargs):
             instance_count=1,
             volume_size_in_gb=100,
             keep_alive_period_in_seconds=None,
+            training_plan_arn=None,
         )
         self.networking = None
         self.stopping_condition = None
diff --git a/sagemaker-train/tests/unit/train/test_serverful_recipe_validation.py b/sagemaker-train/tests/unit/train/test_serverful_recipe_validation.py
index 9e79839959..2c568e63b3 100644
--- a/sagemaker-train/tests/unit/train/test_serverful_recipe_validation.py
+++ b/sagemaker-train/tests/unit/train/test_serverful_recipe_validation.py
@@ -41,6 +41,7 @@ def __init__(self, **kwargs):
             instance_count=4,
             volume_size_in_gb=100,
             keep_alive_period_in_seconds=None,
+            training_plan_arn=None,
         )
         self.networking = None
         self.stopping_condition = None