diff --git a/tests/system/aiplatform/test_e2e_tabular.py b/tests/system/aiplatform/test_e2e_tabular.py
index 0c6967f451..06b6c4d7c2 100644
--- a/tests/system/aiplatform/test_e2e_tabular.py
+++ b/tests/system/aiplatform/test_e2e_tabular.py
@@ -159,7 +159,7 @@ def test_end_to_end_tabular(self, shared_state):
             custom_job._gca_resource.training_task_inputs["scheduling"][
                 "restartJobOnWorkerRestart"
             ]
-            == True
+            is True
         )
 
         custom_prediction = custom_endpoint.predict([_INSTANCE])