diff --git a/src/ntl/run_language_modeling.py b/src/ntl/run_language_modeling.py
index 613cea0..8ca4c9f 100644
--- a/src/ntl/run_language_modeling.py
+++ b/src/ntl/run_language_modeling.py
@@ -8,7 +8,7 @@
 import sys
 import os
 sys.path.append(".")
-os.environ["CUDA_VISIBLE_DEVICES"] = "2"
+os.environ["CUDA_VISIBLE_DEVICES"] = "1"
 
 import time
 import json