diff --git a/spec/classes/chrony_spec.rb b/spec/classes/chrony_spec.rb
index 16aed6b..55dd5d1 100644
--- a/spec/classes/chrony_spec.rb
+++ b/spec/classes/chrony_spec.rb
@@ -9,7 +9,7 @@
         facts
       end
       let(:config_file) do
-        case facts[:osfamily]
+        case facts[:os]['family']
         when 'Archlinux', 'RedHat', 'Suse'
           '/etc/chrony.conf'
         else
@@ -17,7 +17,7 @@
         end
       end
       let(:keys_file) do
-        case facts[:osfamily]
+        case facts[:os]['family']
         when 'Archlinux', 'RedHat', 'Suse'
           '/etc/chrony.keys'
         else