diff --git a/lib/couchbase/model.rb b/lib/couchbase/model.rb
index 2d8ffc0..d272c73 100644
--- a/lib/couchbase/model.rb
+++ b/lib/couchbase/model.rb
@@ -811,7 +811,7 @@ def attributes_with_values
 
     protected :attributes_with_values
 
-    if defined?(::Rails)
+    if defined?(::ActiveModel)
       extend ActiveModel::Callbacks
       extend ActiveModel::Naming
       include ActiveModel::Conversion