diff --git a/lib/puppet_x/puppetlabs/strings/yard/handlers/host_class_handler.rb b/lib/puppet_x/puppetlabs/strings/yard/handlers/host_class_handler.rb
index 06d4a65..982c664 100644
--- a/lib/puppet_x/puppetlabs/strings/yard/handlers/host_class_handler.rb
+++ b/lib/puppet_x/puppetlabs/strings/yard/handlers/host_class_handler.rb
@@ -2,11 +2,11 @@ class PuppetX::PuppetLabs::Strings::YARD::Handlers::HostClassHandler < PuppetX::
   handles HostClassDefinition
 
   process do
-    obj = HostClassObject.new(:root, statement.pops_obj.name) do |o|
-      o.parameters = statement.parameters.map do |a|
-        param_tuple = [a[0].pops_obj.name]
-        param_tuple << ( a[1].nil? ? nil : a[1].source )
-      end
+    obj = HostClassObject.new(:root, statement.pops_obj.name)
+
+    obj.parameters = statement.parameters.map do |a|
+      param_tuple = [a[0].pops_obj.name]
+      param_tuple << ( a[1].nil? ? nil : a[1].source )
     end
     tp = Puppet::Pops::Types::TypeParser.new
     param_type_info = {}