diff --git a/test/integration/targets/prepare_nxos_tests/tasks/main.yml b/test/integration/targets/prepare_nxos_tests/tasks/main.yml
index b0f67b5a9a6..9dfbd13a03d 100644
--- a/test/integration/targets/prepare_nxos_tests/tasks/main.yml
+++ b/test/integration/targets/prepare_nxos_tests/tasks/main.yml
@@ -2,9 +2,9 @@
 # NXAPI is enabled differently on NX-OS platforms.
 # Try both commands to enable NXAPI and ignore any errors.
 - name: enable nxapi on remote device
-  nxos_config:
-    lines:
-      - feature nxapi
+  nxos_feature:
+    feature: nxapi
+    state: enabled
     provider: "{{ cli }}"
   ignore_errors: yes