diff --git a/Dockerfile b/Dockerfile
index 99f381788b0266249e21de59784125e0b32cbfd5..fa500ce52a0e5efb4f127eb8e658e1205e5e03af 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -15,4 +15,4 @@ RUN rm requirements.txt
 # Note that we expect the cloned finn directory on the host to be
 # mounted on /workspace/finn -- see run-docker.sh for an example
 # of how to do this.
-ENV PYTHONPATH "${PYTHONPATH}:/workspace/finn:/workspace/brevitas_cnv_lfc/training_scripts"
+ENV PYTHONPATH "${PYTHONPATH}:/workspace/finn/src:/workspace/brevitas_cnv_lfc/training_scripts"