diff --git a/Dockerfile b/Dockerfile
index 09b987e..f7fe37c 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -14,8 +14,8 @@ RUN apt install -y libfftw3-dev
 RUN apt install -y g++
 RUN apt install -y python3.9
 RUN apt install -y python3-pip
-RUN apt install -y python3-dev
+RUN apt install -y python3.9-dev
 RUN export CXX=$(which g++)
 RUN export CUDACXX=$(which nvcc)
 RUN git submodule update --init --recursive
-RUN python3.9 -m pip install .
+RUN python3 -m pip install .