diff --git a/__tests__/stateProvider.test.ts b/__tests__/stateProvider.test.ts
index bb1b231..315086c 100644
--- a/__tests__/stateProvider.test.ts
+++ b/__tests__/stateProvider.test.ts
@@ -30,12 +30,19 @@ test("StateProvider saves states", async () => {
         .mockImplementation(name =>
             jest.requireActual("@actions/core").getState(name)
         );
+
     const saveStateMock = jest
         .spyOn(core, "saveState")
         .mockImplementation((key, value) => {
             return jest.requireActual("@actions/core").saveState(key, value);
         });
 
+    const setOutputMock = jest
+        .spyOn(core, "setOutput")
+        .mockImplementation((key, value) => {
+            return jest.requireActual("@actions/core").setOutput(key, value);
+        });
+
     const cacheMatchedKey = "node-cache";
 
     const stateProvider: IStateProvider = new StateProvider();
@@ -46,6 +53,7 @@ test("StateProvider saves states", async () => {
 
     expect(getStateMock).toHaveBeenCalledTimes(2);
     expect(saveStateMock).toHaveBeenCalledTimes(2);
+    expect(setOutputMock).toHaveBeenCalledTimes(0);
 });
 
 test("NullStateProvider saves outputs", async () => {
@@ -54,11 +62,19 @@ test("NullStateProvider saves outputs", async () => {
         .mockImplementation(name =>
             jest.requireActual("@actions/core").getState(name)
         );
+
     const setOutputMock = jest
         .spyOn(core, "setOutput")
         .mockImplementation((key, value) => {
             return jest.requireActual("@actions/core").setOutput(key, value);
         });
+
+    const saveStateMock = jest
+        .spyOn(core, "saveState")
+        .mockImplementation((key, value) => {
+            return jest.requireActual("@actions/core").saveState(key, value);
+        });
+
     const cacheMatchedKey = "node-cache";
     const nullStateProvider: IStateProvider = new NullStateProvider();
     nullStateProvider.setState(State.CacheMatchedKey, "outputValue");
@@ -68,4 +84,5 @@ test("NullStateProvider saves outputs", async () => {
 
     expect(getStateMock).toHaveBeenCalledTimes(0);
     expect(setOutputMock).toHaveBeenCalledTimes(2);
+    expect(saveStateMock).toHaveBeenCalledTimes(0);
 });