1 files changed, 2 insertions, 0 deletions
diff --git a/test/rexml/test_document.rb b/test/rexml/test_document.rb
index b99d7b225e..33c62bacc5 100644
--- a/test/rexml/test_document.rb
+++ b/test/rexml/test_document.rb
@@ -32,6 +32,7 @@ EOF
       assert_equal("Hello world!", doc.root.children.first.value)
     end
 
+    class EntityExpansionLimitTest < Test::Unit::TestCase
     XML_WITH_NESTED_ENTITY = <<EOF
 <?xml version="1.0" encoding="UTF-8"?>
 <!DOCTYPE member [
@@ -163,6 +164,7 @@ EOF
     ensure
       REXML::Security.entity_expansion_limit = 10000
     end
+    end
 
     def test_tag_in_cdata_with_not_ascii_only_but_ascii8bit_encoding_source
       tag = "<b>...</b>"