diff --git a/test/index.test.ts b/test/index.test.ts
index e4d149e..148bb01 100644
--- a/test/index.test.ts
+++ b/test/index.test.ts
@@ -62,8 +62,7 @@ test('source image cache pruning', async (t) => {
         multiValueQueryStringParameters: {},
         multiValueHeaders: {},
         isBase64Encoded: false,
-        body: null,
-        netlifyGraphToken: undefined
+        body: null
       },
       {
         functionName: 'ipx',
@@ -75,9 +74,9 @@ test('source image cache pruning', async (t) => {
         logStreamName: '',
         memoryLimitInMB: '',
         getRemainingTimeInMillis: () => 1000,
-        done: () => {},
-        fail: () => {},
-        succeed: () => {}
+        done: () => { },
+        fail: () => { },
+        succeed: () => { }
       }
     )
     if (response) {
@@ -85,7 +84,6 @@ test('source image cache pruning', async (t) => {
     }
   }
 
-
   const cacheSize = readdirSync(join(cacheDir, 'cache')).reduce((acc, filename) => {
     const { size } = statSync(join(cacheDir, 'cache', filename))
     return acc + size