diff --git a/roles/mellanox_drivers/tasks/main.yml b/roles/mellanox_drivers/tasks/main.yml
index 3861de6e08afa8ecac7e76196b95bca3c6aec34e..3091a0cbdf6b9f5f073ccaba9f50a21403f2d1fa 100644
--- a/roles/mellanox_drivers/tasks/main.yml
+++ b/roles/mellanox_drivers/tasks/main.yml
@@ -128,18 +128,6 @@
   become_user: root
   register: service_file
 
-- name: Set failover mode on DGX baremetal nodes
-  lineinfile:
-    path: /etc/modprobe.d/ko2iblnd.conf
-    state: present
-    line: "options ko2iblnd dev_failover=1"
-    insertafter: EOF
-  when:
-   - '"DGX" in ansible_product_name'
-   - '"RedHat" in ansible_distribution'
-  become: true
-  become_user: root 
-
 - name: Reload systemd
   shell: systemctl daemon-reload
   become: true
@@ -166,3 +154,15 @@
 - name: waiting for server to come back
   wait_for_connection: sleep=60 timeout=600 delay=60
   when: reboot_now
+
+- name: Set failover mode on DGX baremetal nodes
+  lineinfile:
+    path: /etc/modprobe.d/ko2iblnd.conf
+    state: present
+    line: "options ko2iblnd dev_failover=1"
+    insertafter: EOF
+  when:
+   - '"DGX" in ansible_product_name'
+   - '"RedHat" in ansible_distribution'
+  become: true
+  become_user: root