lib/rubysmith/templates/%project_name%/lib/%project_path%.rb.erb in rubysmith-5.5.0 vs lib/rubysmith/templates/%project_name%/lib/%project_path%.rb.erb in rubysmith-5.6.0

@@ -1,12 +1,23 @@
 <% if configuration.build_zeitwerk %>
   require "zeitwerk"
   <% if configuration.project_levels.positive? %>
     Zeitwerk::Loader.new.then do |loader|
+      loader.tag = "<%= configuration.project_name %>"
       loader.push_dir "#{__dir__}<%= Array.new(configuration.project_levels, "/..").join %>"
       loader.setup
     end
   <% else %>
-    Zeitwerk::Loader.for_gem.setup
+    Zeitwerk::Loader.new.then do |loader|
+      loader.tag = File.basename __FILE__, ".rb"
+      loader.push_dir __dir__
+      loader.setup
+    end
   <% end %>
 <% end %>
+<% if configuration.build_zeitwerk %>
+<% namespace do %>
+  def self.loader(registry = Zeitwerk::Registry) = registry.loader_for __FILE__
+<% end %>
+<% else %>
 <% namespace %>
+<% end %>