diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py
index 7361068..b87b9b8 100644
--- a/generator/ctrl/ctrl_generator.py
+++ b/generator/ctrl/ctrl_generator.py
@@ -14,7 +14,7 @@ pos_action_starts = ["You attack", "You tell", "You use", "You go"]
 
 # the loss function is a simple categorical crossentropy between the logits and the labels
 def loss(labels, logits):
-    return tf.keras.losses.spae_categorical_crossentropy(labels, logits, from_logits=True)
+    return tf.keras.losses.sparse_categorical_crossentropy(labels, logits, from_logits=True)
 
 
 class CTRLGenerator():
diff --git a/generator/ctrl/install_ctrl.sh b/generator/ctrl/install_ctrl.sh
index 3667751..42cb2df 100755
--- a/generator/ctrl/install_ctrl.sh
+++ b/generator/ctrl/install_ctrl.sh
@@ -20,6 +20,7 @@ cd ..
 curl https://sdk.cloud.google.com | bash
 exec -l $SHELL
 gcloud init
+export CUDA_VISIBLE_DEVICES=0
 
 # Download the 512-length model if specified, 256-length otherwise
 #if [ "$1" = "512" ]