lib/jets/commands/dynamodb/migrator.rb in jets-2.3.0 vs lib/jets/commands/dynamodb/migrator.rb in jets-2.3.1

@@ -18,10 +18,10 @@
 
   def migrate
     path = "#{Jets.root}/#{@path}"
     unless File.exist?(path)
       puts "Unable to find the migration file: #{path}"
-      exit 1 unless ENV['TEST']
+      exit 1 unless Jets.env.test?
     end
 
     require path
     migration_class = get_migration_class
     migration_class.new.up