diff --git a/backend/provider/app.py b/backend/provider/app.py
index 0e91980..654aca3 100644
--- a/backend/provider/app.py
+++ b/backend/provider/app.py
@@ -19,7 +19,6 @@ from procesing.pricers import (
     ElasticityBasedPricer
 )
 from procesing.steps import (
-    StateSpace,
     PredictPricesStep
 )
 from procesing import PipelineContext
diff --git a/docker/Provider.dockerfile b/docker/Provider.dockerfile
index d39da80..6b16413 100644
--- a/docker/Provider.dockerfile
+++ b/docker/Provider.dockerfile
@@ -21,4 +21,6 @@ RUN pip install --no-cache-dir -r requirements.txt
 
 ENV PYTHONPATH=/app:/app/lib:/app/procesing
 
+WORKDIR /app
+
 CMD ["python", "-m", "uvicorn", "provider.app:app", "--host", "0.0.0.0", "--port", "5001"]