Use State objects

al2o3cr · May 31, 2008 · d3c49b6 · d3c49b6
1 parent 30a8d15
commit d3c49b6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/aasm.rb b/lib/aasm.rb
@@ -25,7 +25,7 @@ def aasm_initial_state=(state)
     end
 
     def aasm_state(name, options={})
-      aasm_states << name unless aasm_states.include?(name)
+      aasm_states << AASM::SupportingClasses::State.new(name, options) unless aasm_states.include?(name)
       self.aasm_initial_state = name unless self.aasm_initial_state
 
       define_method("#{name.to_s}?") do
@@ -88,7 +88,7 @@ def aasm_events
     end
 
     def aasm_states_for_select
-      aasm_states.collect { |state| [state.to_s.gsub(/_/, ' ').capitalize, state.to_s] }
+      aasm_states.collect { |state| [state.name.to_s.gsub(/_/, ' ').capitalize, state.name.to_s] }
     end
 
   end