diff --git a/index.ts b/index.ts
index 4e419ec..7ddd84a 100644
--- a/index.ts
+++ b/index.ts
@@ -28,14 +28,14 @@ export default function memoize<A extends unknown[], R extends unknown, T extend
     const id = hash.apply(this, args)
     if (cache.has(id)) return cache.get(id)
     const result = fn.apply(this, args)
+    cache.set(id, result)
     if (result instanceof Promise) {
       // eslint-disable-next-line github/no-then
-      result.catch(error => {
+      return result.catch(error => {
         cache.delete(id)
         throw error
       })
     }
-    cache.set(id, result)
     return result
   }
 }