lib/simple_states.rb in simple_states-1.1.0.rc1 vs lib/simple_states.rb in simple_states-1.1.0.rc2

@@ -61,9 +61,17 @@
 
   def init_state
     self.state = self.class.initial_state if state.nil?
   end
 
+  def state=(state)
+    super(state.to_sym)
+  end
+
+  def state
+    super.try(:to_sym)
+  end
+
   def past_states
     @past_states ||= []
   end
 
   def state?(state, include_past = false)