require activerecord.

ctran · May 12, 2023 · 6720020 · 6720020
1 parent 57b92fe
commit 6720020
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lib/annotate/annotate_routes/header_generator.rb b/lib/annotate/annotate_routes/header_generator.rb
@@ -1,3 +1,4 @@
+require 'active_record'
 require_relative './helpers'
 
 module AnnotateRoutes
@@ -16,7 +17,7 @@ def generate(options = {})
       private
 
       def routes_map(options)
-        command = Rails.version.first.to_i > 5 ? `rails routes` : `rake routes`
+        command = ActiveRecord.version.to_s.first.to_i > 5 ? `rails routes` : `rake routes`
         result = command.chomp("\n").split(/\n/, -1)
 
         # In old versions of Rake, the first line of output was the cwd.  Not so