diff --git a/.drone.yml b/.drone.yml
index 24fac17..c439c7a 100644
--- a/.drone.yml
+++ b/.drone.yml
@@ -13,6 +13,8 @@ steps:
     environment:
       mount_command:
         from_secret: mount_command
+      PYTHONPATH: /workspace/
+      DATA_DIR: /workspace/data/
     commands:
       - python -m pip install -q -U pip
       - pip install -q torchsummary matplotlib pytorch-metric-learning torchinfo torch torchvision tqdm Pillow facenet-pytorch
@@ -22,5 +24,4 @@ steps:
       - mkdir -p data
       - $mount_command
       - ls data/
-      - source setenv.sh
       - CI=True python finetune/facenet_transfer_learning.py
\ No newline at end of file