diff --git a/test/integration/default/controls/package_spec.rb b/test/integration/default/controls/package_spec.rb
index 2a520a6..3414043 100644
--- a/test/integration/default/controls/package_spec.rb
+++ b/test/integration/default/controls/package_spec.rb
@@ -4,11 +4,11 @@
   title 'should be installed'
 
   package_name =
-    case os[:family]
-    when 'debian', 'suse'
-      'vim'
-    else
+    case platform[:family]
+    when 'redhat', 'fedora'
       'vim-enhanced'
+    else
+      'vim'
     end
 
   describe package(package_name) do