diff --git a/python-sdk/pyproject.toml b/python-sdk/pyproject.toml
index 1f493880e..117fbe2d6 100644
--- a/python-sdk/pyproject.toml
+++ b/python-sdk/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "indexify"
-version = "0.2.44"
+version = "0.2.45"
 description = "Python Client for Indexify"
 authors = ["Tensorlake Inc. <support@tensorlake.ai>"]
 license = "Apache 2.0"
diff --git a/python-sdk/tests/test_graph_behaviours.py b/python-sdk/tests/test_graph_behaviours.py
index 1b9226032..5314b9d55 100644
--- a/python-sdk/tests/test_graph_behaviours.py
+++ b/python-sdk/tests/test_graph_behaviours.py
@@ -535,6 +535,10 @@ def my_func_2(input: dict) -> int:
         self.assertEqual(len(output), 1)
         self.assertEqual(output[0], 6)
 
+        output1 = graph.output(invocation_id, my_func.name)
+        self.assertEqual(len(output1), 1)
+        self.assertEqual(output1[0], {"x": 1, "y": 2, "z": 3})
+
     @parameterized.expand([(False), (True)])
     def test_return_dict_args_as_kwargs_in_list(self, is_remote):
         @indexify_function()