diff --git a/README.md b/README.md
index c680e28..f4a33df 100644
--- a/README.md
+++ b/README.md
@@ -59,7 +59,7 @@ To create your first [Report](https://ramppdev.github.io/ablate/modules/report.h
 For example, the built in [Mock](https://ramppdev.github.io/ablate/modules/sources.html#mock-source) can be used to simulate runs:
 
 ```python
-from ablate import sources
+from ablate.sources import Mock
 
 source = Mock(
   grid={"model": ["vgg", "resnet"], "lr": [0.01, 0.001]},
@@ -192,7 +192,7 @@ By default, _ablate_ reports populate blocks based on the global list of runs pa
 To create more complex reports, blocks can be populated with a custom list of runs using the _runs_ parameter:
 
 ```python
-report = ablate.Report(sorted_runs.all())
+report = Report(sorted_runs.all())
 report.add(H1("Report with Sorted Runs and Filtered Runs"))
 report.add(H2("Sorted Runs"))
 report.add(
diff --git a/README.rst b/README.rst
index f6a7010..486e9f6 100644
--- a/README.rst
+++ b/README.rst
@@ -228,7 +228,7 @@ To create more complex reports, blocks can be populated with a custom list of ru
 .. code-block:: python
    :linenos:
 
-   report = ablate.Report(sorted_runs.all())
+   report = Report(sorted_runs.all())
    report.add(H1("Report with Sorted Runs and Filtered Runs"))
    report.add(H2("Sorted Runs"))
    report.add(
diff --git a/ablate/__init__.py b/ablate/__init__.py
index 229aaa1..35b7a33 100644
--- a/ablate/__init__.py
+++ b/ablate/__init__.py
@@ -4,4 +4,4 @@
 
 __all__ = ["blocks", "exporters", "queries", "Report", "sources"]
 
-__version__ = "0.1.2"
+__version__ = "0.2.1"
diff --git a/pyproject.toml b/pyproject.toml
index 43c8f89..32f27fb 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "ablate"
-version = "0.1.2"
+version = "0.2.1"
 description = "ablate turns deep learning experiments into structured, human-readable reports."
 authors = [{ name = "Simon Rampp", email = "simon@rampp.dev" }]
 license = { text = "MIT" }
diff --git a/uv.lock b/uv.lock
index f3b9e0d..b4e712f 100644
--- a/uv.lock
+++ b/uv.lock
@@ -18,7 +18,7 @@ resolution-markers = [
 
 [[package]]
 name = "ablate"
-version = "0.1.2"
+version = "0.2.1"
 source = { editable = "." }
 dependencies = [
     { name = "matplotlib" },