diff --git a/docs/conf.py b/docs/conf.py
index b558159a3..2f12b72c1 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -16,7 +16,7 @@
 project = 'Python SDK reference'
 copyright = '2024, Labelbox'
 author = 'Labelbox'
-release = '6.0.0'
+release = '6.0.1'
 
 # -- General configuration ---------------------------------------------------
 
diff --git a/libs/labelbox/CHANGELOG.md b/libs/labelbox/CHANGELOG.md
index 10cd7fd46..1d5854c73 100644
--- a/libs/labelbox/CHANGELOG.md
+++ b/libs/labelbox/CHANGELOG.md
@@ -1,4 +1,8 @@
 # Changelog
+# Version 6.0.1 (2024-10-30)
+## Fixed
+* Updated labelbox.exception to lbox.exception for IAMIntegration check([#1887](https://github.com/Labelbox/labelbox-python/pull/1887))
+
 # Version 6.0.0 (2024-10-15)
 ## Added
 * Deprecation messages to all export_v2 methods([#1865](https://github.com/Labelbox/labelbox-python/pull/1865))
diff --git a/libs/labelbox/pyproject.toml b/libs/labelbox/pyproject.toml
index a47466004..833925039 100644
--- a/libs/labelbox/pyproject.toml
+++ b/libs/labelbox/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "labelbox"
-version = "6.0.0"
+version = "6.0.1"
 description = "Labelbox Python API"
 authors = [{ name = "Labelbox", email = "engineering@labelbox.com" }]
 dependencies = [
diff --git a/libs/labelbox/src/labelbox/__init__.py b/libs/labelbox/src/labelbox/__init__.py
index 9700f12a9..fc9068c36 100644
--- a/libs/labelbox/src/labelbox/__init__.py
+++ b/libs/labelbox/src/labelbox/__init__.py
@@ -1,6 +1,6 @@
 name = "labelbox"
 
-__version__ = "6.0.0"
+__version__ = "6.0.1"
 
 from labelbox.client import Client
 from labelbox.schema.annotation_import import (
diff --git a/libs/labelbox/src/labelbox/client.py b/libs/labelbox/src/labelbox/client.py
index bcf29665e..288aa6e2b 100644
--- a/libs/labelbox/src/labelbox/client.py
+++ b/libs/labelbox/src/labelbox/client.py
@@ -585,7 +585,7 @@ def create_dataset(
             )
 
             if not validation_result["validateDataset"]["valid"]:
-                raise labelbox.exceptions.LabelboxError(
+                raise LabelboxError(
                     "IAMIntegration was not successfully added to the dataset."
                 )
         except Exception as e:
diff --git a/pyproject.toml b/pyproject.toml
index 7ab0a0b79..6fa5dde92 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -36,7 +36,7 @@ members = ["libs/*", "examples"]
 
 [tool.pytest.ini_options]
 # https://github.com/pytest-dev/pytest-rerunfailures/issues/99
-addopts = "-rP -vvv"
+addopts = "-rP -vvv --reruns 1 --reruns-delay 5 --durations=20 -n auto --cov=labelbox --import-mode=importlib --order-group-scope=module"
 markers = """
     slow: marks tests as slow (deselect with '-m "not slow"')
 """