diff --git a/src/opticalcontroller/Dockerfile b/src/opticalcontroller/Dockerfile
index faea3b2e056768ef9947db108df61928c8a177cb..25805627587f957c341f4e113d0388bf11c73efa 100644
--- a/src/opticalcontroller/Dockerfile
+++ b/src/opticalcontroller/Dockerfile
@@ -67,5 +67,4 @@ COPY src/context/client/. context/client/
 COPY src/opticalcontroller/. opticalcontroller/
 
 # Start the service
-WORKDIR /var/teraflow/opticalcontroller
-ENTRYPOINT ["python", "OpticalController.py"]
+ENTRYPOINT ["python", "-m", "opticalcontroller.OpticalController"]