diff --git a/src/lib.rs b/src/lib.rs
index eca465b..88926ab 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -23,7 +23,7 @@ pub use standalone::*;
 use std::sync::Once;
 pub use utilities::*;
 use whisper_ctx::WhisperInnerContext;
-pub use whisper_ctx_wrapper::WhisperContextWrapper;
+pub use whisper_ctx_wrapper::WhisperContext;
 pub use whisper_ctx::WhisperContextParameters;
 pub use whisper_grammar::{WhisperGrammarElement, WhisperGrammarElementType};
 pub use whisper_params::{FullParams, SamplingStrategy};
diff --git a/src/whisper_ctx_wrapper.rs b/src/whisper_ctx_wrapper.rs
index 51bcdac..b20379f 100644
--- a/src/whisper_ctx_wrapper.rs
+++ b/src/whisper_ctx_wrapper.rs
@@ -3,11 +3,11 @@ use std::sync::Arc;
 
 use crate::{WhisperInnerContext, WhisperContextParameters, WhisperError, WhisperState, WhisperToken};
 
-pub struct WhisperContextWrapper {
+pub struct WhisperContext {
     ctx: Arc<WhisperInnerContext>,
 }
 
-impl WhisperContextWrapper {
+impl WhisperContext {
     fn wrap(ctx: WhisperInnerContext) -> Self {
         Self {
             ctx: Arc::new(ctx),