diff --git a/spec/internal/app/models/category.rb b/spec/internal/app/models/category.rb
index 1a65984..daa1dcf 100644
--- a/spec/internal/app/models/category.rb
+++ b/spec/internal/app/models/category.rb
@@ -1,4 +1,6 @@
 class Category < ActiveRecord::Base
+  include TheModerator::Model
+
   has_one :page
   accepts_nested_attributes_for :page
 end
diff --git a/spec/internal/app/models/page.rb b/spec/internal/app/models/page.rb
index c0cd987..5fb2469 100644
--- a/spec/internal/app/models/page.rb
+++ b/spec/internal/app/models/page.rb
@@ -1,4 +1,6 @@
 class Page < ActiveRecord::Base
+  include TheModerator::Model
+
   has_many :links
   accepts_nested_attributes_for :links
 
diff --git a/spec/internal/db/schema.rb b/spec/internal/db/schema.rb
index 61b621a..1eff192 100644
--- a/spec/internal/db/schema.rb
+++ b/spec/internal/db/schema.rb
@@ -22,6 +22,7 @@ ActiveRecord::Schema.define do
     t.integer  :moderatable_id
     t.string   :moderatable_type
     t.text     :data
+    t.text :data_display
     t.timestamps
   end
 end