diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py
index 30a0c30..4bad39f 100644
--- a/generator/ctrl/ctrl_generator.py
+++ b/generator/ctrl/ctrl_generator.py
@@ -8,7 +8,7 @@ from collections import Counter
 from tensorflow.python import debug as tf_debug
 from tensorflow.python.ops import math_ops
 from tensorflow.python.ops import embedding_ops
-import generator.ctrl.model.fastBPE as fastBPE
+import fastBPE
 
 pos_action_starts = ["You attack", "You tell", "You use", "You go"]
 
diff --git a/generator/ctrl/install_ctrl.sh b/generator/ctrl/install_ctrl.sh
index 42cb2df..99c5d2d 100755
--- a/generator/ctrl/install_ctrl.sh
+++ b/generator/ctrl/install_ctrl.sh
@@ -21,6 +21,7 @@ curl https://sdk.cloud.google.com | bash
 exec -l $SHELL
 gcloud init
 export CUDA_VISIBLE_DEVICES=0
+cd ..
 
 # Download the 512-length model if specified, 256-length otherwise
 #if [ "$1" = "512" ]