diff --git a/scripts/claim.sh b/scripts/claim.sh
index 31b862e..1d0935b 100644
--- a/scripts/claim.sh
+++ b/scripts/claim.sh
@@ -23,6 +23,8 @@ fi
 
 if [ -f ./scripts/claim_mail.sh ]; then
 	bash ./scripts/claim_mail.sh "${JSON}"
+else
+	echo "No email script, not sending."
 fi
 
 bash ./scripts/train.sh "${JSON}"
diff --git a/scripts/train.sh b/scripts/train.sh
index f3f99c0..d20b929 100644
--- a/scripts/train.sh
+++ b/scripts/train.sh
@@ -86,6 +86,11 @@ curl -s -X POST \
 	-F "model=@${MODELPATH}" \
 	"${YOLO_WEB_URL}/job/${ID}"
 
+deactivate
+cd ..
+
 if [ -f ./scripts/completed_mail.sh ]; then
 	bash ./scripts/completed_mail.sh "${JSON}"
+else
+	echo "No email script, not sending"
 fi