diff --git a/ui-tests/package.json b/ui-tests/package.json
index 05a88f43c..e090e4057 100644
--- a/ui-tests/package.json
+++ b/ui-tests/package.json
@@ -13,7 +13,7 @@
   "author": "ipyleaflet",
   "license": "MIT",
   "dependencies": {
-    "@jupyterlab/galata": "^4.3.5",
+    "@jupyterlab/galata": "^5.0.1",
     "@playwright/test": "^1.32.0",
     "klaw-sync": "^6.0.0",
     "rimraf": "^3.0.2"
diff --git a/ui-tests/tests/ipyleaflet.test.ts b/ui-tests/tests/ipyleaflet.test.ts
index bce9cafcd..154927c2f 100644
--- a/ui-tests/tests/ipyleaflet.test.ts
+++ b/ui-tests/tests/ipyleaflet.test.ts
@@ -74,7 +74,7 @@ const testCellOutputs = async (
 test.describe("ipyleaflet Visual Regression", () => {
   test.beforeEach(async ({ page, tmpPath }) => {
     await page.contents.uploadDirectory(
-      path.resolve(__dirname, "../../notebooks"),
+      path.resolve(__dirname, "../notebooks"),
       tmpPath
     );
     await page.filebrowser.openDirectory(tmpPath);