lib/railroady/diagram_graph.rb in railroady-1.0.4 vs lib/railroady/diagram_graph.rb in railroady-1.0.5

@@ -64,11 +64,11 @@
   # Build diagram label
   def dot_label
     return "\t_diagram_info [shape=\"plaintext\", " +
            "label=\"#{@diagram_type} diagram\\l" +
            "Date: #{Time.now.strftime "%b %d %Y - %H:%M"}\\l" +
-           "Migration version: " +
-           "#{ActiveRecord::Migrator.current_version}\\l" +
+           (defined?(ActiveRecord::Migrator) ? "Migration version: " +
+            "#{ActiveRecord::Migrator.current_version}\\l" : "") +
            "Generated by #{APP_HUMAN_NAME} #{APP_VERSION}\\l"+
 		   "http://railroady.prestonlee.com" +
            "\\l\", fontsize=13]\n"
   end