diff --git a/test/integration/context_test.rb b/test/integration/context_test.rb
index 874cc8bd..58f1c331 100644
--- a/test/integration/context_test.rb
+++ b/test/integration/context_test.rb
@@ -24,7 +24,7 @@ class ContextSensitiveDrop < Liquid::Drop
   end
 end
 
-class Category < Liquid::Drop
+class Category
   attr_accessor :name
 
   def initialize(name)
@@ -36,8 +36,9 @@ class Category < Liquid::Drop
   end
 end
 
-class CategoryDrop
+class CategoryDrop < Liquid::Drop
   attr_accessor :category, :context
+
   def initialize(category)
     @category = category
   end