diff --git a/src/whisper_ctx.rs b/src/whisper_ctx.rs
index 9b8b90b..0938821 100644
--- a/src/whisper_ctx.rs
+++ b/src/whisper_ctx.rs
@@ -1,6 +1,6 @@
 use crate::error::WhisperError;
 use crate::whisper_params::FullParams;
-use crate::WhisperToken;
+use crate::{WhisperToken, WhisperTokenData};
 use std::ffi::{c_int, CStr, CString};
 
 /// Safe Rust wrapper around a Whisper context.