lib/blueprints/helper.rb in blueprints-0.3.1 vs lib/blueprints/helper.rb in blueprints-0.3.2

@@ -12,12 +12,12 @@
       Blueprints.delete_tables(*args)
 
       if options[:undo] == :all
         Namespace.root.executed_plans.clear
       else
-        undo = [options[:undo]].flatten.compact
+        undo = [options[:undo]].flatten.compact.collect {|bp| bp.to_s }
         unless (not_found = undo - Namespace.root.executed_plans.to_a).blank?
-          raise(ArgumentError, "Scenario(s) #{not_found} not found")
+          raise(PlanNotFoundError, not_found)
         end
         Namespace.root.executed_plans -= undo
       end
     end
   end
\ No newline at end of file