diff --git a/slurm_qos/tasks/main.yml b/slurm_qos/tasks/main.yml
index abcb446..641fe75 100644
--- a/slurm_qos/tasks/main.yml
+++ b/slurm_qos/tasks/main.yml
@@ -19,8 +19,9 @@
 
     - name: Modify Slurm QoS (if already exists)
       ansible.builtin.command: >
-        sacctmgr -i modify qos {{ item.name }} set
-        {% if item.MaxNodes is defined %}MaxNodes={{ item.MaxNodes }}{% endif %}
+        sacctmgr -i modify qos where name={{ item.name }} set
+        {% if item.MaxTRES is defined %}MaxTRES={{ item.MaxTRES }}{% endif %}
+        {% if item.MaxTRESPU is defined %}MaxTRESPU={{ item.MaxTRESPU }}{% endif %}
         {% if item.MaxSubmitJobs is defined %}MaxSubmitJobs={{ item.MaxSubmitJobs }}{% endif %}
         {% if item.MaxJobsPerUser is defined %}MaxJobsPerUser={{ item.MaxJobsPerUser }}{% endif %}
         {% if item.GraceTime is defined %}GraceTime={{ item.GraceTime }}{% endif %}