diff --git a/brainglobe_segmentation/atlas/utils.py b/brainglobe_segmentation/atlas/utils.py
index b3e2d4c..9d5dd08 100644
--- a/brainglobe_segmentation/atlas/utils.py
+++ b/brainglobe_segmentation/atlas/utils.py
@@ -36,7 +36,8 @@ def structure_from_viewer(coordinates, atlas_layer, atlas):
                     Layer, which contains the annotation / region
                     information for every structure in the (registered)
                     atlas
-    atlas         : Brainglobe atlas (bg_atlasapi.bg_atlas.BrainGlobeAtlas)
+    atlas         : Brainglobe atlas
+                    (brainglobe_atlasapi.bg_atlas.BrainGlobeAtlas)
 
     Returns
     -------
diff --git a/pyproject.toml b/pyproject.toml
index 0591e56..c2521a4 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -18,7 +18,7 @@ classifiers = [
 ]
 requires-python = ">=3.9"
 dependencies = [
-    "bg-atlasapi",
+    "brainglobe-atlasapi >=2.0.1",
     "brainglobe-napari-io >=0.3.0",
     "brainglobe-utils >=0.4.0",
     "napari >=0.4.5",
@@ -80,9 +80,11 @@ line-length = 79
 [tool.ruff]
 line-length = 79
 exclude = ["__init__.py", "build", ".eggs"]
-select = ["I", "E", "F"]
 fix = true
 
+[tool.ruff.lint]
+select = ["I", "E", "F"]
+
 [tool.tox]
 legacy_tox_ini = """
 # For more information about tox, see https://tox.readthedocs.io/en/latest/
diff --git a/tests/conftest.py b/tests/conftest.py
index 0000534..02a4a62 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -2,7 +2,7 @@
 from pathlib import Path
 
 import pytest
-from bg_atlasapi import BrainGlobeAtlas
+from brainglobe_atlasapi import BrainGlobeAtlas
 
 from brainglobe_segmentation.segment import SegmentationWidget