diff --git a/rainfallwrangler/slurm-jsonl2tfrecord.job b/rainfallwrangler/slurm-jsonl2tfrecord.job
index 5519fe5..2f0b164 100755
--- a/rainfallwrangler/slurm-jsonl2tfrecord.job
+++ b/rainfallwrangler/slurm-jsonl2tfrecord.job
@@ -5,7 +5,7 @@
 #SBATCH --gres=gpu:1
 #SBATCH -o %j.%N.%a.out
 #SBATCH -e %j.%N.%a.err
-#SBATCH -p gpu05,gpu
+#SBATCH -p compute
 #SBATCH --time=3-00:00:00
 
 module load utilities/multi