diff --git a/spec/classes/puppet_agent_spec.rb b/spec/classes/puppet_agent_spec.rb
index 872ba34b3..066f71dc5 100644
--- a/spec/classes/puppet_agent_spec.rb
+++ b/spec/classes/puppet_agent_spec.rb
@@ -122,6 +122,7 @@ def global_facts(facts, os)
 
             it { is_expected.to compile.with_all_deps }
             it { expect { catalogue }.not_to raise_error }
+            it { is_expected.to contain_class('puppet-agent').with_package_version(version) }
           end
         end