diff --git a/zschema/tests.py b/zschema/tests.py
index 85855d6..31769d4 100644
--- a/zschema/tests.py
+++ b/zschema/tests.py
@@ -115,6 +115,7 @@ def test_invalid(self):
     {
         "type": "INTEGER",
         "name": "ip",
+        "doc": "The IP Address of the host",
         "mode": "NULLABLE"
     },
 ]
@@ -154,16 +155,16 @@ def setUp(self):
 
         heartbleed = SubRecord({
             "heartbeat_support":Boolean(),
-            "heartbleed_vulnerable":Boolean(),
+            "heartbleed_vulnerable":Boolean(category="Vulnerabilities"),
             "timestamp":DateTime()
         })
         self.host = Record({
-                "ipstr":IPv4Address(required=True),
-                "ip":Long(),
+                "ipstr":IPv4Address(required=True, examples=["8.8.8.8"]),
+                "ip":Long(doc="The IP Address of the host"),
                 Port(443):SubRecord({
                     "tls":String(),
                     "heartbleed":heartbleed
-                }),
+                }, category="heartbleed"),
                 "tags":ListOf(String())
         })