diff --git a/javascript/packages/orchestrator/src/test-runner/assertions.ts b/javascript/packages/orchestrator/src/test-runner/assertions.ts
index 56e047c2f..3e3f87f5d 100644
--- a/javascript/packages/orchestrator/src/test-runner/assertions.ts
+++ b/javascript/packages/orchestrator/src/test-runner/assertions.ts
@@ -1,4 +1,4 @@
-import { ApiPromise, Keyring } from "@polkadot/api";
+import { ApiPromise, Keyring, WsProvider } from "@polkadot/api";
 import { decorators, isValidHttpUrl } from "@zombienet/utils";
 import { assert, expect } from "chai";
 import execa from "execa";
@@ -291,6 +291,7 @@ const CustomJs = ({
     (global as any).zombie = {
       ApiPromise,
       Keyring,
+      WsProvider,
       util: utilCrypto,
       connect,
       registerParachain,