lib/stateflow.rb in stateflow-0.0.3 vs lib/stateflow.rb in stateflow-0.0.4

@@ -1,9 +1,9 @@
 module Stateflow
   def self.included(base)
     base.send :include, InstanceMethods
-    Stateflow::Persistence.set(base)
     base.extend ClassMethods
+    Stateflow::Persistence.set(base)
   end
   
   def self.persistence
     @@persistence ||= :active_record
   end