lib/importmap/engine.rb in importmap-rails-0.3.4 vs lib/importmap/engine.rb in importmap-rails-0.4.0

@@ -1,21 +1,18 @@
 require "importmap/map"
 
 module Importmap
   class Engine < ::Rails::Engine
-    config.importmap = Importmap::Map.new
-
+    config.importmap = Importmap::Map.new.draw("config/importmap.rb")
     config.autoload_once_paths = %W( #{root}/app/helpers )
 
     initializer "importmap.reloader" do |app|
       app.config.paths.add "config/importmap.rb"
 
-      reloader = Importmap::Reloader.new
-
-      reloader.execute
-      app.reloaders << reloader
-      app.reloader.to_run do
+      Importmap::Reloader.new.tap do |reloader|
         reloader.execute
+        app.reloaders << reloader
+        app.reloader.to_run { reloader.execute }
       end
     end
 
     initializer "importmap.assets" do
       if Rails.application.config.respond_to?(:assets)