diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index d3f5ea72b..99ad9cc02 100644
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -20,7 +20,7 @@ jobs:
       - name: Restore cache
         run: |
           cache-util restore cargo_git cargo_registry yarn_cache
-          cache-util restore aurora-engine-target@generic@${{ hashFiles('**/Cargo.lock') }}:target
+          cache-util restore aurora-engine-target@${{ github.base.ref }}@${{ hashFiles('**/Cargo.lock') }}:target
       - name: Build contracts
         run: cargo make build-contracts
       - name: Test contracts
@@ -40,7 +40,7 @@ jobs:
       - name: Save cache
         run: |
           cache-util save cargo_git cargo_registry yarn_cache
-          cache-util msave aurora-engine-target@generic@${{ hashFiles('**/Cargo.lock') }}:target
+          cache-util msave aurora-engine-target@${{ github.base.ref }}@${{ hashFiles('**/Cargo.lock') }}:target
 env:
   CARGO_TERM_COLOR: always
   CARGO_INCREMENTAL: 0