diff --git a/PyTorch/Segmentation/nnUNet/run_benchmark_single_node.sh b/PyTorch/Segmentation/nnUNet/run_benchmark_single_node.sh
index 5e632665b34c2a9ad6985483e6ff9a4df089a91e..485eff950e2aac9a759ced4722ca1eeffba669b6 100644
--- a/PyTorch/Segmentation/nnUNet/run_benchmark_single_node.sh
+++ b/PyTorch/Segmentation/nnUNet/run_benchmark_single_node.sh
@@ -19,11 +19,6 @@ cat <<EOT >  $SBATCH_DIR
 #SBATCH --output=$SBATCH_OUT_DIR
 ##SBATCH --reservation=devel
 
-MODULE_NAME=nnunet_for_pytorch
-MODULE_VERSION=21.11.0
-WORK_DIR=/proj/nsc_testing/xuan/berzelius-benchmarks/PyTorch/Segmentation/nnUNet
-CONTAINER_DIR=/proj/nsc_testing/xuan/containers/${MODULE_NAME}_${MODULE_VERSION}.sif
-
 rm -f $WORK_DIR/results/$LOG_DIR
 apptainer exec --nv -B ${WORK_DIR}/data:/data -B ${WORK_DIR}/results:/results --pwd /workspace/nnunet_pyt $CONTAINER_DIR python scripts/benchmark.py --mode ${5} --gpus ${3} --dim ${1} --batch_size ${4} --amp --logname='$LOG_DIR'