diff --git a/spec/acceptance/observium_install_spec.rb b/spec/acceptance/observium_install_spec.rb
index fa1e1fb..0efeb3c 100644
--- a/spec/acceptance/observium_install_spec.rb
+++ b/spec/acceptance/observium_install_spec.rb
@@ -1,6 +1,7 @@
 # frozen_string_literal: true
 
 require 'spec_helper_acceptance'
+require 'rspec-puppet-facts'
 
 describe 'Installation', if: ['centos', 'redhat', 'ubuntu'].include?(os[:family]) do
     before(:all) do
diff --git a/spec/spec_helper_acceptance_local.rb b/spec/spec_helper_acceptance_local.rb
index 6cb462a..fd17284 100644
--- a/spec/spec_helper_acceptance_local.rb
+++ b/spec/spec_helper_acceptance_local.rb
@@ -1,5 +1,11 @@
 include PuppetLitmus
 
 def install_packge(package)
-    run_shell("yum -y install #{package}")
+    if :osfamily == 'Ubuntu' do
+        run_shell("apt -y install #{package}")
+    end
+    if :osfamily == 'RedHat' do
+        run_shell("yum -y install #{package}")
+    end
+    
 end
\ No newline at end of file