diff --git a/dvc.yaml b/dvc.yaml
index 891946e..dfc92bd 100644
--- a/dvc.yaml
+++ b/dvc.yaml
@@ -1,11 +1,11 @@
 stages:
   train:
-    cmd: python ./train.py 10
+    cmd: python3 ./train.py 10
     deps:
     - Customers.csv
     - train.py
   test:
-    cmd: python ./test.py
+    cmd: python3 ./test.py
     deps:
     - cifar_net.pth
     - test.py