diff --git a/tests/test_cli.py b/tests/test_cli.py
index 72705fc..d9d2683 100644
--- a/tests/test_cli.py
+++ b/tests/test_cli.py
@@ -126,6 +126,7 @@ def test_push_pipeline(self, mock_upload_pipeline, mock_get_pipeline, mock_graph
             mock_pipeline = MagicMock(spec=Pipeline)
             mock_pipeline.code = pipeline_name
             mock_get_pipeline.return_value = mock_pipeline
+            mock_upload_pipeline.return_value = {"versionName": version}
 
             result = self.runner.invoke(pipelines_push, [tmp, "--name", version])
             self.assertEqual(result.exit_code, 0)