diff --git a/spec/classes/params_spec.rb b/spec/classes/params_spec.rb
index 8544532..0da9dfc 100644
--- a/spec/classes/params_spec.rb
+++ b/spec/classes/params_spec.rb
@@ -5,7 +5,7 @@
   shared_examples 'osfamily' do |family|
     let(:facts) {{ :osfamily => family }}
 
-    it { should include_class('smartd::params') }
+    it { should contain_class('smartd::params') }
   end
 
   describe 'for osfamily RedHat' do
@@ -29,7 +29,7 @@
     end
   
     it 'should fail' do
-      expect { should include_class('smartd::params') }.
+      expect { should contain_class('smartd::params') }.
         to raise_error(Puppet::Error, /not supported on Solaris/)
     end
   end
diff --git a/spec/classes/smartd_spec.rb b/spec/classes/smartd_spec.rb
index 126ba5c..9297c81 100644
--- a/spec/classes/smartd_spec.rb
+++ b/spec/classes/smartd_spec.rb
@@ -299,8 +299,8 @@
       end
   
       it do
-        should include_class('smartd')
-        should include_class('smartd::params')
+        should contain_class('smartd')
+        should contain_class('smartd::params')
         should contain_package('smartmontools')
         should contain_service('smartd')
         should contain_file('/etc/smartd.conf')\