diff --git a/packages/element-web-playwright-common/package.json b/packages/element-web-playwright-common/package.json
index 87610dd81e..01cd478cb6 100644
--- a/packages/element-web-playwright-common/package.json
+++ b/packages/element-web-playwright-common/package.json
@@ -12,18 +12,8 @@
     "engines": {
         "node": ">=20.0.0"
     },
-    "exports": {
-        ".": {
-            "import": "./lib/index.js",
-            "require": "./lib/index.js",
-            "types": "./lib/index.d.ts"
-        },
-        "./stale-screenshot-reporter": {
-            "import": "./lib/stale-screenshot-reporter.js",
-            "require": "./lib/stale-screenshot-reporter.js",
-            "types": "./lib/stale-screenshot-reporter.d.ts"
-        }
-    },
+    "main": "lib/index.js",
+    "types": "lib/index.d.ts",
     "bin": {
         "playwright-screenshots": "playwright-screenshots.sh"
     },
diff --git a/packages/element-web-playwright-common/src/testcontainers/HomeserverContainer.ts b/packages/element-web-playwright-common/src/testcontainers/HomeserverContainer.ts
index ccb242767b..18bc613aeb 100644
--- a/packages/element-web-playwright-common/src/testcontainers/HomeserverContainer.ts
+++ b/packages/element-web-playwright-common/src/testcontainers/HomeserverContainer.ts
@@ -5,7 +5,7 @@ SPDX-License-Identifier: AGPL-3.0-only OR LicenseRef-Element-Commercial
 Please see LICENSE files in the repository root for full details.
 */
 
-import { type AbstractStartedContainer, type GenericContainer } from "testcontainers";
+import { type AbstractStartedContainer, type TestContainer } from "testcontainers";
 import { type APIRequestContext, type TestInfo } from "@playwright/test";
 
 import { type StartedMatrixAuthenticationServiceContainer } from "./mas";
@@ -41,7 +41,7 @@ export interface HomeserverInstance {
     setThreepid(userId: string, medium: string, address: string): Promise<void>;
 }
 
-export interface HomeserverContainer<Config> extends GenericContainer {
+export interface HomeserverContainer<Config> extends TestContainer {
     /**
      * Set a configuration field in the config
      * @param key - the key to set