diff --git a/tests/websocket_test.ts b/tests/websocket_test.ts
index 22655f2..3d5760b 100644
--- a/tests/websocket_test.ts
+++ b/tests/websocket_test.ts
@@ -3,8 +3,8 @@ const { I} = inject();
 Feature('Web socket');
 
 Scenario('Web Socket', async () => {
-    await I.amOnPage('https://websocketstest.com/');
+    I.amOnPage('https://websocketstest.com/');
     I.waitForText('Work for You!');
     const wsMessages = await I.grabWebSocketMessages();
-    await I.expectContain(wsMessages, 'RECEIVED ID');
+    I.expectContain(wsMessages, 'RECEIVED ID');
 });
\ No newline at end of file