diff --git a/test/integration/codelens.test.ts b/test/integration/codelens.test.ts
index f3c07d9..e3e1b8f 100644
--- a/test/integration/codelens.test.ts
+++ b/test/integration/codelens.test.ts
@@ -194,8 +194,8 @@
 		const found = [] as string[];
 		for (let i = 0; i < codeLenses.length; i++) {
 			const lens = codeLenses[i];
-			if (lens.command.command === 'go.test.cursor') {
-				found.push(lens.command.arguments[0].functionName);
+			if (lens.command?.command === 'go.test.cursor') {
+				found.push(lens.command.arguments?.[0].functionName);
 			}
 		}
 		found.sort();