lib/dry/system/rails/railtie.rb in dry-system-rails-0.3.0 vs lib/dry/system/rails/railtie.rb in dry-system-rails-0.3.1

@@ -20,15 +20,10 @@
           set_or_reload(:Container, container)
           set_or_reload(:Import, container.injector)
 
           container.refresh_boot_files if reloading?
 
-          container.finalize!(freeze: freeze?)
-        end
-
-        # @api private
-        def freeze?
-          !::Rails.env.test?
+          container.finalize! unless ::Rails.env.test?
         end
 
         # @api private
         def reloading?
           app_namespace.const_defined?(:Container)