diff --git a/src/Umbraco.Web.UI.Client/src/backoffice/documents/documents/workspace/actions/save-and-preview.action.ts b/src/Umbraco.Web.UI.Client/src/backoffice/documents/documents/workspace/actions/save-and-preview.action.ts
index de24db885a..71b4ce30d7 100644
--- a/src/Umbraco.Web.UI.Client/src/backoffice/documents/documents/workspace/actions/save-and-preview.action.ts
+++ b/src/Umbraco.Web.UI.Client/src/backoffice/documents/documents/workspace/actions/save-and-preview.action.ts
@@ -1,14 +1,10 @@
 import { UmbWorkspaceAction } from '../../../../shared/components/workspace/workspace-action';
 import { UmbDocumentWorkspaceContext } from '../document-workspace.context';
-import { UmbDocumentRepository } from '../../repository/document.repository';
 import { UmbControllerHostInterface } from '@umbraco-cms/controller';
 
-export class UmbDocumentSaveAndPreviewWorkspaceAction extends UmbWorkspaceAction<
-	UmbDocumentRepository,
-	UmbDocumentWorkspaceContext
-> {
-	constructor(host: UmbControllerHostInterface, repositoryAlias: string) {
-		super(host, repositoryAlias);
+export class UmbDocumentSaveAndPreviewWorkspaceAction extends UmbWorkspaceAction<UmbDocumentWorkspaceContext> {
+	constructor(host: UmbControllerHostInterface) {
+		super(host);
 	}
 
 	async execute() {
@@ -17,6 +13,6 @@ export class UmbDocumentSaveAndPreviewWorkspaceAction extends UmbWorkspaceAction
 		const document = this.workspaceContext.getData();
 		// TODO: handle errors
 		if (!document) return;
-		this.repository?.saveAndPreview();
+		this.workspaceContext.repository?.saveAndPreview();
 	}
 }