diff --git a/src/Umbraco.Web.UI.Client/src/core/mocks/domains/document-type-tree.handlers.ts b/src/Umbraco.Web.UI.Client/src/core/mocks/domains/document-type-tree.handlers.ts
index 155bfac2c6..0524138e3f 100644
--- a/src/Umbraco.Web.UI.Client/src/core/mocks/domains/document-type-tree.handlers.ts
+++ b/src/Umbraco.Web.UI.Client/src/core/mocks/domains/document-type-tree.handlers.ts
@@ -30,7 +30,7 @@ export const handlers = [
 		const keys = req.params.keys as string;
 		if (!keys) return;
 
-		const items = umbDocumentTypeData.getTreeItem(keys);
+		const items = umbDocumentTypeData.getTreeItem(keys.split(','));
 
 		return res(ctx.status(200), ctx.json(items));
 	}),