code cleanup

neo4jrb · Dec 29, 2023 · 1f9a470 · 1f9a470
1 parent 214170d
commit 1f9a470
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/active_graph/railtie.rb b/lib/active_graph/railtie.rb
@@ -8,8 +8,8 @@
   require 'rails/generators/named_base'
   require 'rails/railtie'
   require File.expand_path('../rails/generators/migration_helper.rb', __dir__)
+  Rails::Generators::GeneratedAttribute.include ActiveGraph::Generators::GeneratedAttribute
 end
-Rails::Generators::GeneratedAttribute.include ActiveGraph::Generators::GeneratedAttribute if defined?(Rails)
 
 module ActiveGraph
   class Railtie < ::Rails::Railtie