diff --git a/test/webdriverio/test/basic_test.ts b/test/webdriverio/test/basic_test.ts
index 61d59d4b..83e02304 100644
--- a/test/webdriverio/test/basic_test.ts
+++ b/test/webdriverio/test/basic_test.ts
@@ -275,8 +275,8 @@ suite('Keyboard navigation on Blocks', function () {
       'text_print_1',
     );
     chai.assert.equal(
-      await getCurrentFocusNodeId(this.browser),
-      Blockly.ASTNode.types.NEXT,
+      await getFocusedConnectionType(this.browser),
+      Blockly.ConnectionType.NEXT_STATEMENT,
     );
   });
 });