diff --git a/src/whisper_ctx.rs b/src/whisper_ctx.rs
index eaa8302..330391e 100644
--- a/src/whisper_ctx.rs
+++ b/src/whisper_ctx.rs
@@ -551,7 +551,7 @@ impl WhisperContextParameters {
     pub fn new() -> Self {
         Self::default()
     }
-    pub fn use_gpu(mut self, use_gpu: bool) -> Self {
+    pub fn use_gpu(&mut self, use_gpu: bool) -> &mut Self {
         self.use_gpu = use_gpu;
         self
     }