diff --git a/.drone.yml b/.drone.yml
index c439c7a..2ca7c02 100644
--- a/.drone.yml
+++ b/.drone.yml
@@ -13,8 +13,8 @@ steps:
     environment:
       mount_command:
         from_secret: mount_command
-      PYTHONPATH: /workspace/
-      DATA_DIR: /workspace/data/
+      PYTHONPATH: /drone/src/
+      DATA_DIR: /drone/src/data/
     commands:
       - python -m pip install -q -U pip
       - pip install -q torchsummary matplotlib pytorch-metric-learning torchinfo torch torchvision tqdm Pillow facenet-pytorch