diff --git a/generate/generate_test.go b/generate/generate_test.go
index 892693f..c0dbcba 100644
--- a/generate/generate_test.go
+++ b/generate/generate_test.go
@@ -231,9 +231,9 @@ func TestGenerateRun(test *testing.T) {
 		}
 		testEncode(
 			&messageConnect,
-			tu.S(0xC1).AddVar(
-				[]byte { 0x00, 0x00, 0x66, 'r', 'a', 'r', 'i', 't', 'y' }, // TODO tag and payload
-				[]byte { 0x00, 0x01, 0x64, 'g', 'e', 'm', 's' }, // TODO tag and payload
+			tu.S(0xC1, 0x02).AddVar(
+				[]byte { 0x00, 0x00, 0x66, 'r', 'a', 'r', 'i', 't', 'y' },
+				[]byte { 0x00, 0x01, 0x64, 'g', 'e', 'm', 's' },
 				))
 	`)
 }