lib/generators/test_unit/policy_generator.rb in pundit-1.1.0 vs lib/generators/test_unit/policy_generator.rb in pundit-2.0.0.beta1

@@ -1,8 +1,8 @@
 module TestUnit
   module Generators
     class PolicyGenerator < ::Rails::Generators::NamedBase
-      source_root File.expand_path(File.join(File.dirname(__FILE__), 'templates'))
+      source_root File.expand_path('templates', __dir__)
 
       def create_policy_test
         template 'policy_test.rb', File.join('test/policies', class_path, "#{file_name}_policy_test.rb")
       end
     end