lib/aasm/aasm.rb in aasm-3.0.2 vs lib/aasm/aasm.rb in aasm-3.0.3

@@ -1,13 +1,13 @@
 module AASM
 
   def self.included(base) #:nodoc:
     base.extend AASM::ClassMethods
 
-    AASM::Persistence.set_persistence(base)
     unless AASM::StateMachine[base]
       AASM::StateMachine[base] = AASM::StateMachine.new('')
     end
+    AASM::Persistence.set_persistence(base)
    super
   end
 
   module ClassMethods
     def inherited(klass)