diff --git a/src/main/modules/nhentai/nhentai-ipc-controller.ts b/src/main/modules/nhentai/nhentai-ipc-controller.ts
index ad206d6..0335bb6 100644
--- a/src/main/modules/nhentai/nhentai-ipc-controller.ts
+++ b/src/main/modules/nhentai/nhentai-ipc-controller.ts
@@ -1,7 +1,6 @@
 import path from 'path';
 import { createWriteStream } from 'fs-extra';
 import { container } from '../../core/container';
-import type { Work } from '../../entities/library/work';
 import type { DialogInterface } from '../dialog/dialog-interface';
 import { answer } from '../ipc/annotations/answer';
 import type { SourceGetterInterface } from '../source/source-getter-interface';
@@ -50,7 +49,7 @@ export class NhentaiIpcController implements IpcController {
   }
 
   @answer(IpcChannel.NHENTAI_GET_WORK)
-  public async nhentaiGetWork({ galleryId }: { galleryId: string }): Promise<Work> {
+  public async nhentaiGetWork({ galleryId }: { galleryId: string }): Promise<WorkEntityInterface> {
     const work = await this.nhentaiSourceGetter.find(galleryId);
 
     return work;