diff --git a/src/llama.cpp b/src/llama.cpp
index e0299b950..1df81eb9c 100644
--- a/src/llama.cpp
+++ b/src/llama.cpp
@@ -14122,7 +14122,7 @@ static void llama_set_inputs(llama_context & lctx, const llama_batch & batch) {
 
             if (lctx.inp_KQ_mask) {
                 GGML_ASSERT(ggml_backend_buffer_is_host(lctx.inp_KQ_mask->buffer));
-                data_swa = (float *) lctx.inp_KQ_mask->data;
+                data = (float *) lctx.inp_KQ_mask->data;
             }
 
             if (lctx.inp_KQ_mask_swa) {