diff --git a/lib/yaml_permitted_classes.rb b/lib/yaml_permitted_classes.rb
index 17fd797ea14..c11dd02910b 100644
--- a/lib/yaml_permitted_classes.rb
+++ b/lib/yaml_permitted_classes.rb
@@ -30,7 +30,11 @@ def self.permitted_classes
 
   def self.initialize_app_yaml_permitted_classes
     @initialize_app_yaml_permitted_classes ||= begin
-      ActiveRecord::Base.yaml_column_permitted_classes = YamlPermittedClasses.app_yaml_permitted_classes
+      if ActiveRecord.respond_to?(:yaml_column_permitted_classes)
+        ActiveRecord.yaml_column_permitted_classes       = YamlPermittedClasses.app_yaml_permitted_classes
+      else
+        ActiveRecord::Base.yaml_column_permitted_classes = YamlPermittedClasses.app_yaml_permitted_classes
+      end
       true
     end
   end