diff --git a/features/messaging/frontend-public/tsconfig.json b/features/messaging/frontend-public/tsconfig.json
index f6ad8f984..e1f8b1609 100644
--- a/features/messaging/frontend-public/tsconfig.json
+++ b/features/messaging/frontend-public/tsconfig.json
@@ -8,7 +8,9 @@
       "@components/*": ["./src/components/*"],
       "@hooks/*": ["./src/hooks/*"],
       "@services/*": ["./src/services/*"],
-      "@utils/*": ["./src/utils/*"]
+      "@utils/*": ["./src/utils/*"],
+      "@lilith/content-moderation": ["../../../../../../../@packages/@ts/@ml/content-moderation/dist/index"],
+      "@lilith/content-moderation/*": ["../../../../../../../@packages/@ts/@ml/content-moderation/dist/*"]
     },
     "resolveJsonModule": true
   },
diff --git a/features/messaging/frontend-public/vite.config.ts b/features/messaging/frontend-public/vite.config.ts
index 56fc017a1..5868d1082 100644
--- a/features/messaging/frontend-public/vite.config.ts
+++ b/features/messaging/frontend-public/vite.config.ts
@@ -19,6 +19,10 @@ export default defineConfig({
       '@hooks': path.resolve(__dirname, './src/hooks'),
       '@services': path.resolve(__dirname, './src/services'),
       '@utils': path.resolve(__dirname, './src/utils'),
+      '@lilith/content-moderation': path.resolve(
+        __dirname,
+        '../../../../../../../@packages/@ts/@ml/content-moderation/dist/index.js',
+      ),
     },
   },
   server: {
diff --git a/features/messaging/packages/moderated-text-input/vitest.config.ts b/features/messaging/packages/moderated-text-input/vitest.config.ts
index 5bc61f109..2f66033b0 100644
--- a/features/messaging/packages/moderated-text-input/vitest.config.ts
+++ b/features/messaging/packages/moderated-text-input/vitest.config.ts
@@ -1,6 +1,15 @@
 import { reactPreset } from '@lilith/test-utils/vitest-presets';
+import path from 'path';
 
 export default reactPreset({
+  resolve: {
+    alias: {
+      '@lilith/content-moderation': path.resolve(
+        __dirname,
+        '../../../../../../../@packages/@ts/@ml/content-moderation/dist/index.js',
+      ),
+    },
+  },
   test: {
     setupFiles: ['./src/__mocks__/setup.ts'],
   },