diff --git a/src/dfcx_scrapi/core/environments.py b/src/dfcx_scrapi/core/environments.py
index c9ee2c21..bc54e3c9 100644
--- a/src/dfcx_scrapi/core/environments.py
+++ b/src/dfcx_scrapi/core/environments.py
@@ -309,10 +309,10 @@ def update_environment(
         if environment_obj:
             env = environment_obj
             env.name = environment_id
-            mask = self._update_kwargs(environment_obj)
+            mask = self._update_kwargs(env)
         elif kwargs:
             env = self.get_environment(environment_id)
-            mask = self._update_kwargs(environment_obj, **kwargs)
+            mask = self._update_kwargs(env, **kwargs)
 
         request = types.environment.UpdateEnvironmentRequest(
             environment=env, update_mask=mask)
diff --git a/src/dfcx_scrapi/core/test_cases.py b/src/dfcx_scrapi/core/test_cases.py
index f6dd04df..154f8ab6 100644
--- a/src/dfcx_scrapi/core/test_cases.py
+++ b/src/dfcx_scrapi/core/test_cases.py
@@ -462,12 +462,12 @@ def update_test_case(
         if obj:
             test_case = obj
             test_case.name = test_case_id
-            mask = self._update_kwargs(obj)
+            mask = self._update_kwargs(test_case)
         elif kwargs:
             if not test_case_id:
                 test_case_id = self.test_case_id
             test_case = self.get_test_case(test_case_id)
-            mask = self._update_kwargs(obj, **kwargs)
+            mask = self._update_kwargs(test_case, **kwargs)
 
         request = types.test_case.UpdateTestCaseRequest(
             test_case=test_case, update_mask=mask)