diff --git a/py-denormalized/src/context.rs b/py-denormalized/src/context.rs
index 7e2372f..e6e6ffa 100644
--- a/py-denormalized/src/context.rs
+++ b/py-denormalized/src/context.rs
@@ -53,11 +53,8 @@ impl PyContext {
     #[new]
     pub fn new(py: Python) -> PyResult<Self> {
         let rt = &get_tokio_runtime(py).0;
-        let fut: JoinHandle<denormalized::common::error::Result<Context>> = rt.spawn(async move {
-            Ok(Context::new()?
-                .with_slatedb_backend(String::from("/tmp/checkpoints/simple-agg-checkpoint-1"))
-                .await)
-        });
+        let fut: JoinHandle<denormalized::common::error::Result<Context>> =
+            rt.spawn(async move { Ok(Context::new()?) });
 
         let context = wait_for_future(py, fut).map_err(py_denormalized_err)??;
 
diff --git a/py-denormalized/src/utils.rs b/py-denormalized/src/utils.rs
index a7f41f6..2fc9732 100644
--- a/py-denormalized/src/utils.rs
+++ b/py-denormalized/src/utils.rs
@@ -6,7 +6,7 @@ use tokio::runtime::Runtime;
 
 /// Utility to get the Tokio Runtime from Python
 pub(crate) fn get_tokio_runtime(py: Python) -> PyRef<TokioRuntime> {
-    let datafusion = py.import_bound("denormalized._internal").unwrap();
+    let datafusion = py.import_bound("denormalized._d_internal").unwrap();
     let tmp = datafusion.getattr("runtime").unwrap();
     match tmp.extract::<PyRef<TokioRuntime>>() {
         Ok(runtime) => runtime,