diff --git a/evaluate.py b/evaluate.py
index a1956493..9d20b43f 100644
--- a/evaluate.py
+++ b/evaluate.py
@@ -4,9 +4,9 @@ from __future__ import absolute_import, division, print_function
 
 if __name__ == "__main__":
     try:
-        from coqui_stt_training import evaluate as ds_evaluate
+        from coqui_stt_training import evaluate as stt_evaluate
     except ImportError:
         print("Training package is not installed. See training documentation.")
         raise
 
-    ds_evaluate.main()
+    stt_evaluate.main()
diff --git a/train.py b/train.py
index f878f72c..7477ed7f 100755
--- a/train.py
+++ b/train.py
@@ -4,9 +4,9 @@ from __future__ import absolute_import, division, print_function
 
 if __name__ == "__main__":
     try:
-        from coqui_stt_training import train as ds_train
+        from coqui_stt_training import train as stt_train
     except ImportError:
         print("Training package is not installed. See training documentation.")
         raise
 
-    ds_train.main()
+    stt_train.main()