diff --git a/install.bat b/install.bat
index 10d27fe..b53d2e2 100644
--- a/install.bat
+++ b/install.bat
@@ -3,7 +3,7 @@ setlocal enabledelayedexpansion
 
 :: First check if `install.bat` (this) has needed files in same directory
 if not exist %~dp0\yolo.py ( echo `yolo.py` missing in %~dp0 cannot install & goto :choice_default_3 )
-if not exist %~dp0\prompt.txt ( echo `prompt.txt` missing in %~dp0 cannot install & goto :choice_default_3 ) 
+if not exist %~dp0\yolo.prompt ( echo `yolo.prompt` missing in %~dp0 cannot install & goto :choice_default_3 ) 
 if not exist %~dp0\yolo.yaml ( echo `yolo.yaml` missing in %~dp0 cannot install & goto :choice_default_3 ) 
 
 
diff --git a/install.sh b/install.sh
index f2bd104..17fe451 100755
--- a/install.sh
+++ b/install.sh
@@ -7,7 +7,7 @@ TARGET_FULLPATH=$TARGET_DIR/yolo.py
 mkdir -p $TARGET_DIR
 
 echo "- Copying files..."
-cp yolo.py prompt.txt yolo.yaml $TARGET_DIR
+cp yolo.py yolo.prompt yolo.yaml $TARGET_DIR
 chmod +x $TARGET_FULLPATH
 
 # Creates two aliases for use