diff --git a/api-server/server/utils/in-memory-cache.test.js b/api-server/server/utils/in-memory-cache.test.js
index a16c85d8ff..572e88e024 100644
--- a/api-server/server/utils/in-memory-cache.test.js
+++ b/api-server/server/utils/in-memory-cache.test.js
@@ -36,9 +36,7 @@ describe('InMemoryCache', () => {
     it('can handle promises correctly', done => {
       const cache = inMemoryCache(before, reportErrorStub);
       const promisedUpdate = () => new Promise(resolve => resolve(after));
-      cache.update(promisedUpdate);
-      // because async
-      setImmediate(() => {
+      cache.update(promisedUpdate).then(() => {
         expect(cache.get()).toBe(after);
         done();
       });