diff --git a/__tests__/cache-restore.test.ts b/__tests__/cache-restore.test.ts
index d84c2296..182cf226 100644
--- a/__tests__/cache-restore.test.ts
+++ b/__tests__/cache-restore.test.ts
@@ -183,20 +183,28 @@ virtualenvs.path = "{cache-dir}/virtualenvs"  # /Users/patrick/Library/Caches/py
             cachePaths
           );
         }
+        
+        const restoredKeys = restoreCacheSpy.mock.results.map(result => result.value);
 
-        if (process.platform === 'linux' && packageManager === 'pip') {
-          expect(infoSpy).toHaveBeenCalledWith(
-            `Cache restored from key: setup-python-${process.env['RUNNER_OS']}-${process.arch}-20.04-Ubuntu-python-${pythonVersion}-${packageManager}-${fileHash}`
-          );
-        } else if (packageManager === 'poetry') {
-          expect(infoSpy).toHaveBeenCalledWith(
-            `Cache restored from key: setup-python-${process.env['RUNNER_OS']}-${process.arch}-python-${pythonVersion}-${packageManager}-v2-${fileHash}`
-          );
-        } else {
-          expect(infoSpy).toHaveBeenCalledWith(
-            `Cache restored from key: setup-python-${process.env['RUNNER_OS']}-${process.arch}-python-${pythonVersion}-${packageManager}-${fileHash}`
-          );
-        }
+        restoredKeys.forEach(restoredKey => {
+          if (restoredKey) {
+            if (process.platform === 'linux' && packageManager === 'pip') {
+              expect(infoSpy).toHaveBeenCalledWith(
+                `Cache restored from key: setup-python-${process.env['RUNNER_OS']}-${process.arch}-20.04-Ubuntu-python-${pythonVersion}-${packageManager}-${fileHash}`
+              );
+            } else if (packageManager === 'poetry') {
+              expect(infoSpy).toHaveBeenCalledWith(
+                `Cache restored from key: setup-python-${process.env['RUNNER_OS']}-${process.arch}-python-${pythonVersion}-${packageManager}-v2-${fileHash}`
+              );
+            } else {
+              expect(infoSpy).toHaveBeenCalledWith(
+                `Cache restored from key: setup-python-${process.env['RUNNER_OS']}-${process.arch}-python-${pythonVersion}-${packageManager}-${fileHash}`
+              );
+            }
+          } else {
+            expect(infoSpy).toHaveBeenCalledWith(`${packageManager} cache is not found`);
+          }
+        });
       },
       30000
     );
diff --git a/__tests__/data/requirements.txt b/__tests__/data/requirements.txt
index fa40ab3c..9f6e6cba 100644
--- a/__tests__/data/requirements.txt
+++ b/__tests__/data/requirements.txt
@@ -34,7 +34,7 @@ pillow==7.2
 
 pygments==2.6.1
 
-pyinstaller==3.6
+pyinstaller==5.31.1
 
 pyparsing==2.4.7; python_version >= '2.6' and python_version not in '3.0, 3.1, 3.2'