diff --git a/instant-segment-py/Cargo.toml b/instant-segment-py/Cargo.toml
index 55690ce..899ff93 100644
--- a/instant-segment-py/Cargo.toml
+++ b/instant-segment-py/Cargo.toml
@@ -18,5 +18,5 @@ crate-type = ["cdylib"]
 [dependencies]
 bincode = "1.3.2"
 instant-segment = { version = "0.11", path = "../instant-segment", features = ["with-serde"] }
-pyo3 = { version = "0.21", features = ["extension-module"] }
+pyo3 = { version = "0.22", features = ["extension-module"] }
 smartstring = "1"
diff --git a/instant-segment-py/src/lib.rs b/instant-segment-py/src/lib.rs
index 7dd6d15..d72a72c 100644
--- a/instant-segment-py/src/lib.rs
+++ b/instant-segment-py/src/lib.rs
@@ -6,7 +6,7 @@ use std::io::{BufReader, BufWriter};
 
 use pyo3::exceptions::PyValueError;
 use pyo3::pybacked::PyBackedStr;
-use pyo3::types::{PyAnyMethods, PyIterator, PyModule};
+use pyo3::types::{PyAnyMethods, PyIterator, PyModule, PyModuleMethods};
 use pyo3::{pyclass, pymethods, pymodule, Bound};
 use pyo3::{PyErr, PyRef, PyRefMut, PyResult, Python};
 use smartstring::alias::String as SmartString;