1 files changed, 3 insertions, 2 deletions
diff --git a/src/routine.cc b/src/routine.cc
index e0a75e41..cd4d82fb 100644
--- a/src/routine.cc
+++ b/src/routine.cc
@@ -96,8 +96,9 @@ StatusCode Routine<T>::SetUp() {
       }
       if (build_status == BuildStatus::kInvalid) { return StatusCode::kInvalidBinary; }
 
-      // Store the compiled program in the cache (atomic for thread-safety)
-      StoreProgramToCache(program);
+      // Store the compiled kernel in the cache
+      auto binary = program.GetIR();
+      StoreBinaryToCache(binary);
     } catch (...) { return StatusCode::kBuildProgramFailure; }
   }