diff --git a/setup.py b/setup.py
index 317679d..58aaf29 100644
--- a/setup.py
+++ b/setup.py
@@ -10,7 +10,7 @@
 
 setup(
     name="weavel",
-    version="1.9.0",
+    version="1.9.1",
     packages=find_namespace_packages(),
     entry_points={},
     description="Weavel, Prompt Optimization and Evaluation for LLM Applications",
diff --git a/weavel/client.py b/weavel/client.py
index 3c9edb6..54c7d76 100644
--- a/weavel/client.py
+++ b/weavel/client.py
@@ -13,14 +13,13 @@
 from dotenv import load_dotenv
 from weavel._worker import Worker
 
-# from weavel.types.instances import Session, Span, Trace
 from weavel.object_clients import (
     GenerationClient,
     SessionClient,
     SpanClient,
     TraceClient,
 )
-from weavel.types.datasets import Dataset, DatasetItem, Prompt, PromptVersion
+from weavel.types import Dataset, DatasetItem, Prompt, PromptVersion
 
 load_dotenv()