diff --git a/test/streams/flatten.js b/test/streams/flatten.js
index 429b39f..9b6e17e 100644
--- a/test/streams/flatten.js
+++ b/test/streams/flatten.js
@@ -48,6 +48,21 @@ test('flatten - classdesc', function (t) {
   });
 });
 
+test('flatten - augments', function (t) {
+  evaluate(function () {
+    /** @augments Foo */
+    return 0;
+  }, function (result) {
+    t.deepEqual(result[0].augments, [{
+      'title': 'augments',
+      'description': null,
+      'type': null,
+      'name': 'Foo'
+    }]);
+    t.end();
+  });
+});
+
 test('flatten - kind', function (t) {
   evaluate(function () {
     /** @kind class */
diff --git a/test/streams/hierarchy.js b/test/streams/hierarchy.js
index de1f0f0..ce500ae 100644
--- a/test/streams/hierarchy.js
+++ b/test/streams/hierarchy.js
@@ -65,3 +65,21 @@ test('hierarchy', function (t) {
     t.end();
   });
 });
+
+test('hierarchy - missing memberof', function (t) {
+  evaluate(function () {
+
+    /**
+     * Get foo
+     * @memberof DoesNotExist
+     * @returns {Number} foo
+     */
+    function getFoo () {
+      return this.foo;
+    }
+
+  }, function (result) {
+    t.equal(result.length, 1);
+    t.end();
+  });
+});