diff --git a/Dockerfile b/Dockerfile
index 00fd3f8e..74158e6b 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,15 +1,8 @@
 FROM colmap/colmap:latest
 MAINTAINER Paul-Edouard Sarlin
-ARG PYTHON_VERSION=3.8
-RUN apt-get update -y
-RUN apt-get install -y unzip wget software-properties-common
-RUN add-apt-repository ppa:deadsnakes/ppa && \
-    apt-get -y update && \
-    apt-get install -y python${PYTHON_VERSION}
-RUN wget https://bootstrap.pypa.io/get-pip.py && python${PYTHON_VERSION} get-pip.py
-RUN update-alternatives --install /usr/bin/python3 python3 /usr/bin/python${PYTHON_VERSION} 1
+RUN apt-get update -y && apt-get install -y unzip wget python3-pip
 COPY . /app
 WORKDIR app/
 RUN pip3 install --upgrade pip
 RUN pip3 install -r requirements.txt
-RUN pip3 install notebook
+RUN pip3 install --upgrade notebook ipywidgets