diff --git a/substrafl/remote/register/register.py b/substrafl/remote/register/register.py
index d9bbb0f0..55745fb8 100644
--- a/substrafl/remote/register/register.py
+++ b/substrafl/remote/register/register.py
@@ -43,8 +43,7 @@
 RUN apt-get update -y\
     && pip uninstall -y setuptools\
     && apt-get install -y {binary_dependencies}\
-    && apt-get clean\
-    && rm -rf /var/lib/apt/lists/*
+    && apt-get clean
 """
 
 _GPU_BASE_IMAGE = """
diff --git a/tests/dependency/test_dependency.py b/tests/dependency/test_dependency.py
index a8b30f87..660d4591 100644
--- a/tests/dependency/test_dependency.py
+++ b/tests/dependency/test_dependency.py
@@ -307,7 +307,8 @@ def test_binary_dependencies(
 
         client = network.clients[0]
         algo_deps = Dependency(
-            binary_dependencies=["gcc"],
+            pypi_dependencies=["pytest"],
+            binary_dependencies=["python3-dev"],
             editable_mode=True,
         )
         function_key = self._register_function(dummy_algo_class(), algo_deps, client, session_dir)