diff --git a/lib/aasm.rb b/lib/aasm.rb
index 711706c..f747b5d 100644
--- a/lib/aasm.rb
+++ b/lib/aasm.rb
@@ -12,5 +12,6 @@ require 'aasm/core/state'
 require 'aasm/localizer'
 require 'aasm/state_machine'
 require 'aasm/persistence'
+require 'aasm/persistence/base'
 require 'aasm/persistence/plain_persistence'
 require 'aasm/aasm'
diff --git a/lib/aasm/persistence.rb b/lib/aasm/persistence.rb
index 1ae3b4a..fd49c67 100644
--- a/lib/aasm/persistence.rb
+++ b/lib/aasm/persistence.rb
@@ -10,16 +10,16 @@ module AASM
           require_persistence(:active_record)
           include_persistence base, :active_record
         elsif hierarchy.include?("Mongoid::Document")
-          require_persistence(:active_record)
+          require_persistence(:mongoid)
           include_persistence base, :mongoid
         elsif hierarchy.include?("MongoMapper::Document")
-          require_persistence(:active_record)
+          require_persistence(:mongo_mapper)
           include_persistence base, :mongo_mapper
         elsif hierarchy.include?("Sequel::Model")
-          require_persistence(:active_record)
+          require_persistence(:sequel)
           include_persistence base, :sequel
         elsif hierarchy.include?("Dynamoid::Document")
-          require_persistence(:active_record)
+          require_persistence(:dynamoid)
           include_persistence base, :dynamoid
         elsif hierarchy.include?("CDQManagedObject")
           include_persistence base, :core_data_query