diff --git a/modules/deeplearning/cloudheight.py b/modules/deeplearning/cloudheight.py
index 44b7088c207e8eea6fb04e2370614425ea8246b7..9f28ae68283b3719ddf275f1925bbc979961c1d3 100644
--- a/modules/deeplearning/cloudheight.py
+++ b/modules/deeplearning/cloudheight.py
@@ -844,7 +844,7 @@ class CloudHeightNN:
 
     def run(self, matchup_dict, train_dict=None, valid_dict=None):
         #with self.strategy.scope():
-        with tf.device('/device:GPU:'+self.gpu_device):
+        with tf.device('/device:GPU:'+str(self.gpu_device)):
             self.setup_pipeline(matchup_dict, train_dict=train_dict, valid_test_dict=valid_dict)
             self.build_model()
             self.build_training()