diff --git a/flojoy/flojoy_node_venv.py b/flojoy/flojoy_node_venv.py
index d6b104c..c949675 100644
--- a/flojoy/flojoy_node_venv.py
+++ b/flojoy/flojoy_node_venv.py
@@ -279,10 +279,9 @@ def decorator(func, *, pip_dependencies=pip_dependencies, verbose=verbose):
             package.name: package.version
             for package in importlib.metadata.distributions()
         }
-        # TODO(roulbac): remove the git ref
         pip_dependencies = sorted(
             [
-                f"git+https://git@github.com/flojoy-ai/python.git@reda-stream-logs-all",
+                f"flojoy=={packages_dict['flojoy']}",
                 f"cloudpickle=={packages_dict['cloudpickle']}",
             ]
             + pip_dependencies