diff --git a/tests/bgp_capability_test.c b/tests/bgp_capability_test.c
index 96f18f0..3cfc198 100644
--- a/tests/bgp_capability_test.c
+++ b/tests/bgp_capability_test.c
@@ -311,7 +311,16 @@
     { 0x41, 0x4, 0xab, 0xcd, 0xef, 0x12 }, /* AS: 2882400018 */
     6, SHOULD_PARSE, 2882400018,
   },
-  /* 20 */
+  { "AS4",
+    "AS4 capability: short",
+    { 0x41, 0x4, 0xab, 0xcd, 0xef }, /* AS: 2882400018 */
+    5, SHOULD_ERR,
+  },
+  { "AS4",
+    "AS4 capability: long",
+    { 0x41, 0x4, 0xab, 0xcd, 0xef, 0x12, 0x12 },
+    7, SHOULD_ERR, 2882400018,
+  },
   { "GR",
     "GR capability",
     { /* hdr */		CAPABILITY_CODE_RESTART, 0xe,
@@ -328,7 +337,6 @@
     },
     16, SHOULD_PARSE,
   },
-  /* 21 */
   { "GR-short",
     "GR capability, but header length too short",
     { /* hdr */		0x40, 0xa,
@@ -343,9 +351,8 @@
       /* safi */	0x2,
       /* flags */	0x1,
     },
-    16, SHOULD_PARSE,
+    15 /* array is 16 though */, SHOULD_ERR,
   },
-  /* 22 */
   { "GR-long",
     "GR capability, but header length too long",
     { /* hdr */		0x40, 0xf,
@@ -358,6 +365,7 @@
       /* flags */	0x0,
       /* afi */		0x0, 0x2,
       /* safi */	0x2,
+      /* flags */	0x01,
     },
     16, SHOULD_ERR,
   },