diff --git a/src/cpp/src/model_runner.hpp b/src/cpp/src/model_runner.hpp
index 68254844f4..bf06c9e315 100644
--- a/src/cpp/src/model_runner.hpp
+++ b/src/cpp/src/model_runner.hpp
@@ -173,7 +173,7 @@ class ModelRunner {
         _set_block_indices(sequence_groups, scheduler_output, total_num_blocks);
 
         if (!m_cache_rotation_coefficients.empty()) {
-            _set_cache_rotation_coefficients();
+            _set_cache_rotation_coefficients(sequence_groups, scheduler_output);
         }
 
         m_request.set_tensor("block_indices_begins", block_indices_begins);