diff --git a/flow/templates/slurm.sh b/flow/templates/slurm.sh
index 88f4c274..596b3d6c 100644
--- a/flow/templates/slurm.sh
+++ b/flow/templates/slurm.sh
@@ -4,9 +4,11 @@
 #!/bin/bash
 #SBATCH --job-name="{{ id }}"
         {% set memory_requested = operations | calc_memory(parallel)  %}
+{% block memory scoped %}
         {% if memory_requested %}
 #SBATCH --mem={{ memory_requested|format_memory }}
         {% endif %}
+{% endblock memory %}
         {% if partition %}
 #SBATCH --partition={{ partition }}
         {% endif %}