diff --git a/Dockerfile b/Dockerfile
index 8663296..61b7370 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -6,6 +6,7 @@ RUN : \
     build-essential \
     libjpeg-dev \
     zlib1g-dev \
+    python3.6 \
     python3-dev \
     python3-pip \
     python3-venv \
diff --git a/setup.py b/setup.py
index f4a98b5..7d4dd09 100644
--- a/setup.py
+++ b/setup.py
@@ -5,6 +5,7 @@
     version="0.1",
     packages=["hls_vi"],
     install_requires=[
+        "dataclasses",
         "matplotlib",
         "numpy~=1.19.0",
         "rasterio",