lib/machinist.rb in elight-machinist-1.0.0 vs lib/machinist.rb in elight-machinist-1.0.1

@@ -75,10 +75,10 @@
       end
     end
   end
 end
 
-if RAILS_ENV == 'test'
+if Module.const_defined?("RAILS_ENV") && RAILS_ENV == 'test'
   require 'sham'
  
   class ActiveRecord::Base
     include Machinist::ActiveRecordExtensions
   end