From c34705542d735041cf3acbda5a03fee10542b641 Mon Sep 17 00:00:00 2001
From: Alexis <alexis.challande@trailofbits.com>
Date: Tue, 1 Oct 2024 15:53:19 +0200
Subject: [PATCH] Fix types

---
 rust/src/lib.rs  | 2 ++
 rust/src/util.rs | 4 ++--
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/rust/src/lib.rs b/rust/src/lib.rs
index e571567..25e9683 100644
--- a/rust/src/lib.rs
+++ b/rust/src/lib.rs
@@ -57,6 +57,7 @@ impl TimeStampReq {
 
     #[getter]
     fn message_imprint(&self) -> PyResult<PyMessageImprint> {
+    // fn message_imprint<'p>(&self, py: pyo3::Python<'p>) -> PyResult<PyMessageImprint> {
         Ok(PyMessageImprint {
             contents: OwnedMessageImprint::try_new(Arc::clone(&self.raw), |v| {
                 Ok::<_, ()>(v.borrow_dependent().message_imprint.clone())
@@ -80,6 +81,7 @@ impl TimeStampReq {
 self_cell::self_cell!(
     struct OwnedMessageImprint {
         owner: Arc<OwnedTimeStamReq>,
+        // owner: pyo3::Py<pyo3::types::PyBytes>,
         #[covariant]
         dependent: RawMessageImprint,
     }
diff --git a/rust/src/util.rs b/rust/src/util.rs
index 452a1b2..c1f11f1 100644
--- a/rust/src/util.rs
+++ b/rust/src/util.rs
@@ -64,8 +64,8 @@ pub(crate) fn datetime_to_py_utc<'p>(
     py: pyo3::Python<'p>,
     dt: &asn1::DateTime,
 ) -> pyo3::PyResult<pyo3::Bound<'p, pyo3::PyAny>> {
-    let timezone = types::DATETIME_TIMEZONE_UTC.get(py)?;
-    types::DATETIME_DATETIME.get(py)?.call1((
+    let timezone = DATETIME_TIMEZONE_UTC.get(py)?;
+    DATETIME_DATETIME.get(py)?.call1((
         dt.year(),
         dt.month(),
         dt.day(),