diff --git a/src/distrib.slurm b/src/distrib.slurm
index f203a4619e83230a5be180b4727b9b66d7864044..1a86c33303c0c549b5d0ca5fe461e778b90d0173 100644
--- a/src/distrib.slurm
+++ b/src/distrib.slurm
@@ -6,7 +6,7 @@
 #SBATCH --cpus-per-task=4
 #SBATCH --output=output.%j
 #SBATCH --error=err.%j
-#SBATCH --time=00:10:00
+#SBATCH --time=00:20:00
 #SBATCH --partition=develbooster
 #SBATCH --gres=gpu:4                                                                                                                                                                                                                           
 
@@ -17,4 +17,4 @@ cd $HOME/2023-may-intro-to-supercompting-jsc/src
 source sc_venv_template/activate.sh
 
 # Run the demo
-time srun python distrib.py
+time srun accelerate launch distrib.py