lib/beaker-hiera.rb in beaker-hiera-0.1.1 vs lib/beaker-hiera.rb in beaker-hiera-0.2.0

@@ -1,11 +1,15 @@
+require 'beaker-hiera/helpers'
+require 'beaker-hiera/version'
+
 module Beaker
   module DSL
     module Helpers
       module Hiera
-        require 'stringify-hash'
-        require 'beaker-hiera/helpers'
-        require 'beaker-hiera/version'
+        include Beaker::DSL::Hiera::Helpers
       end
     end
   end
 end
+
+# Register the DSL extension
+Beaker::DSL.register( Beaker::DSL::Helpers::Hiera )