diff --git a/test/integration/targets/vyos_interface/tests/cli/basic.yaml b/test/integration/targets/vyos_interface/tests/cli/basic.yaml
index acb60a92ef3..3832b0c72be 100644
--- a/test/integration/targets/vyos_interface/tests/cli/basic.yaml
+++ b/test/integration/targets/vyos_interface/tests/cli/basic.yaml
@@ -4,7 +4,7 @@
 - name: Run vyos lsmod command
   vyos_command:
     commands:
-      - lsmod | grep virtio
+      - lsmod
   register: lsmod_out
 
 - name: Set up - delete interface
@@ -20,8 +20,7 @@
     speed: 100
     duplex: half
     mtu: 256
-  when: "'virtio' not in lsmod_out.stdout"
-
+  when: "'virtio_net' not in lsmod_out.stdout[0]"
   register: result
 
 - assert:
@@ -31,7 +30,7 @@
       - '"set interfaces ethernet eth1 speed 100" in result.commands'
       - '"set interfaces ethernet eth1 duplex half" in result.commands'
       - '"set interfaces ethernet eth1 mtu 256" in result.commands'
-  when: "'virtio' not in lsmod_out.stdout"
+  when: "'virtio_net' not in lsmod_out.stdout[0]"
 
 - name: Configure interface params (idempotent)
   vyos_interface:
@@ -42,12 +41,12 @@
     duplex: half
     mtu: 256
   register: result
-  when: "'virtio' not in lsmod_out.stdout"
+  when: "'virtio_net' not in lsmod_out.stdout[0]"
 
 - assert:
     that:
       - 'result.changed == false'
-  when: "'virtio' not in lsmod_out.stdout"
+  when: "'virtio' not in lsmod_out.stdout[0]"
 
 - name: Change interface params
   vyos_interface:
@@ -58,7 +57,7 @@
     duplex: full
     mtu: 512
   register: result
-  when: "'virtio' not in lsmod_out.stdout"
+  when: "'virtio_net' not in lsmod_out.stdout[0]"
 
 - assert:
     that:
@@ -67,7 +66,7 @@
       - '"set interfaces ethernet eth1 speed 1000" in result.commands'
       - '"set interfaces ethernet eth1 duplex full" in result.commands'
       - '"set interfaces ethernet eth1 mtu 512" in result.commands'
-  when: "'virtio' not in lsmod_out.stdout"
+  when: "'virtio_net' not in lsmod_out.stdout[0]"
 
 - name: Disable interface
   vyos_interface:
@@ -124,7 +123,7 @@
       - { name: eth1, description: test-interface-1,  speed: 100, duplex: half, mtu: 512}
       - { name: eth2, description: test-interface-2,  speed: 1000, duplex: full, mtu: 256}
   register: result
-  when: "'virtio' not in lsmod_out.stdout"
+  when: "'virtio_net' not in lsmod_out.stdout[0]"
 
 - assert:
     that:
@@ -137,7 +136,7 @@
       - '"set interfaces ethernet eth2 speed 1000" in result.commands'
       - '"set interfaces ethernet eth2 duplex full" in result.commands'
       - '"set interfaces ethernet eth2 mtu 256" in result.commands'
-  when: "'virtio' not in lsmod_out.stdout"
+  when: "'virtio_net' not in lsmod_out.stdout[0]"
 
 - name: Set interface on aggregate (idempotent)
   vyos_interface:
@@ -145,12 +144,12 @@
       - { name: eth1, description: test-interface-1,  speed: 100, duplex: half, mtu: 512}
       - { name: eth2, description: test-interface-2,  speed: 1000, duplex: full, mtu: 256}
   register: result
-  when: "'virtio' not in lsmod_out.stdout"
+  when: "'virtio_net' not in lsmod_out.stdout[0]"
 
 - assert:
     that:
       - 'result.changed == false'
-  when: "'virtio' not in lsmod_out.stdout"
+  when: "'virtio_net' not in lsmod_out.stdout[0]"
 
 - name: Disable interface on aggregate
   vyos_interface: