diff --git a/release/docker/memgraph_community.dockerfile b/release/docker/memgraph_community.dockerfile
index c6278d136..0b7149522 100644
--- a/release/docker/memgraph_community.dockerfile
+++ b/release/docker/memgraph_community.dockerfile
@@ -3,9 +3,12 @@ FROM debian:stretch
 ARG deb_release
 
 RUN apt-get update && apt-get install -y \
-    openssl libcurl3 libssl1.1 python3 libpython3.5 \
+    openssl libcurl3 libssl1.1 python3 libpython3.5 python3-pip \
+    --no-install-recommends \
   && rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/*
 
+RUN pip3 install networkx
+
 COPY ${deb_release} /
 
 # Install memgraph package
diff --git a/release/docker/memgraph_enterprise.dockerfile b/release/docker/memgraph_enterprise.dockerfile
index 0405cbc63..375b04b49 100644
--- a/release/docker/memgraph_enterprise.dockerfile
+++ b/release/docker/memgraph_enterprise.dockerfile
@@ -3,9 +3,12 @@ FROM debian:stretch
 ARG deb_release
 
 RUN apt-get update && apt-get install -y \
-    openssl libcurl3 libssl1.1 libseccomp2 python3 libpython3.5 \
+    openssl libcurl3 libssl1.1 libseccomp2 python3 libpython3.5 python3-pip \
+    --no-install-recommends \
   && rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/*
 
+RUN pip3 install networkx
+
 COPY ${deb_release} /
 
 # Install memgraph package