diff --git a/__tests__/cache-restore.test.ts b/__tests__/cache-restore.test.ts
index d1c26967..cb37cc93 100644
--- a/__tests__/cache-restore.test.ts
+++ b/__tests__/cache-restore.test.ts
@@ -121,7 +121,7 @@ virtualenvs.path = "{cache-dir}/virtualenvs"  # /Users/patrick/Library/Caches/py
             }
           });
 
-          computeKeysSpy.mockImplementation(() => 'Ubuntu-20.4');
+          computeKeysSpy.mockResolvedValue('Ubuntu-20.4');
         }
 
         await cacheDistributor.restoreCache();