2 files changed, 10 insertions, 2 deletions
diff --git a/spec/handlers/c/source/class_handler_spec.rb b/spec/handlers/c/source/class_handler_spec.rb
index bfac3b6..ac97bc8 100644
--- a/spec/handlers/c/source/class_handler_spec.rb
+++ b/spec/handlers/c/source/class_handler_spec.rb
@@ -8,7 +8,11 @@ describe YARD::MRuby::Handlers::C::Source::ClassHandler do
   end
 
   it "should register classes under namespaces" do
-    parse_init 'cFoo = mrb_define_class_under(mrb, cBar, "Foo", rb_cObject);'
+    parse_init(<<-eof)
+      cBar = mrb_define_class(mrb, "Bar", mrb->object_class);
+      cFoo = mrb_define_class_under(mrb, cBar, "Foo", rb_cObject);'
+    eof
+
     expect(Registry.at('Bar::Foo').type).to be :class
   end
 
diff --git a/spec/handlers/c/source/module_handler_spec.rb b/spec/handlers/c/source/module_handler_spec.rb
index fd08ae6..9641eb3 100644
--- a/spec/handlers/c/source/module_handler_spec.rb
+++ b/spec/handlers/c/source/module_handler_spec.rb
@@ -7,7 +7,11 @@ describe YARD::MRuby::Handlers::C::Source::ModuleHandler do
   end
 
   it "should register modules under namespaces" do
-    parse_init 'mFoo = mrb_define_module_under(mrb, mBar, "Foo");'
+    parse_init(<<-eof)
+      cBar = mrb_define_module(mrb,"Bar");
+      cFoo = mrb_define_module_under(mrb, cBar, "Foo");
+    eof
+
     expect(Registry.at('Bar::Foo').type).to be :module
   end