diff --git a/handlers/main.yml b/handlers/main.yml
index 7bc3819..ca0f8e4 100644
--- a/handlers/main.yml
+++ b/handlers/main.yml
@@ -3,8 +3,7 @@
 - name: update grub
   command: "{{ grub_update_grub_command }}"
   when:
-    - ansible_virtualization_type != "docker"
-    - ansible_virtualization_type != "container"
+    - ansible_connection != "docker"
   notify: 
     - reboot