diff --git a/src/lib.rs b/src/lib.rs
index ed17306..70b6d46 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -14,6 +14,7 @@ pub use standalone::*;
 pub use utilities::*;
 pub use whisper_ctx::WhisperContext;
 pub use whisper_ctx::WhisperContextParameters;
+pub use whisper_grammar::{WhisperGrammarElement, WhisperGrammarElementType};
 pub use whisper_params::{FullParams, SamplingStrategy};
 pub use whisper_state::WhisperState;
 
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
     }
diff --git a/sys/Cargo.toml b/sys/Cargo.toml
index baae8d1..b429860 100644
--- a/sys/Cargo.toml
+++ b/sys/Cargo.toml
@@ -15,15 +15,21 @@ include = [
     "whisper.cpp/CMakeLists.txt",
     "whisper.cpp/ggml.c",
     "whisper.cpp/ggml.h",
-    "whisper.cpp/ggml-opencl.c",
-    "whisper.cpp/ggml-opencl.h",
+    "whisper.cpp/ggml-alloc.c",
+    "whisper.cpp/ggml-alloc.h",
+    "whisper.cpp/ggml-backend.c",
+    "whisper.cpp/ggml-backend.h",
+    "whisper.cpp/ggml-backend-impl.h",
     "whisper.cpp/ggml-cuda.cu",
     "whisper.cpp/ggml-cuda.h",
-    "whisper.cpp/ggml-alloc.h",
-    "whisper.cpp/ggml-alloc.c",
-    "whisper.cpp/ggml-metal.metal",
-    "whisper.cpp/ggml-metal.m",
+    "whisper.cpp/ggml-impl.h",
     "whisper.cpp/ggml-metal.h",
+    "whisper.cpp/ggml-metal.m",
+    "whisper.cpp/ggml-metal.metal",
+    "whisper.cpp/ggml-opencl.cpp",
+    "whisper.cpp/ggml-opencl.h",
+    "whisper.cpp/ggml-quants.h",
+    "whisper.cpp/ggml-quants.c",
     "whisper.cpp/LICENSE",
     "whisper.cpp/whisper.cpp",
     "whisper.cpp/whisper.h",