diff --git a/api/server/services/Files/Code/__tests__/process-traversal.spec.js b/api/server/services/Files/Code/__tests__/process-traversal.spec.js
index b7c62e3d79..d36cd5c7af 100644
--- a/api/server/services/Files/Code/__tests__/process-traversal.spec.js
+++ b/api/server/services/Files/Code/__tests__/process-traversal.spec.js
@@ -139,7 +139,7 @@ describe('processCodeOutput path traversal protection', () => {
   });
 
   test('getRetentionExpiry is called with the request object', async () => {
-    mockSanitizeFilename.mockReturnValueOnce('output.csv');
+    mockSanitizeArtifactPath.mockReturnValueOnce('output.csv');
     await processCodeOutput({ ...baseParams, name: 'output.csv' });
     expect(getRetentionExpiry).toHaveBeenCalledWith(baseParams.req);
   });