diff --git a/setup.py b/setup.py
index c2630095..89a17ad7 100644
--- a/setup.py
+++ b/setup.py
@@ -25,7 +25,7 @@ def _setup_install_requires() -> List:
     return ["torch>=1.7.0", "transformers<=4.40", "pydantic<2.7"]
 
 def _setup_extras() -> Dict:
-    return {"dev": ["black==22.12.0", "isort==5.8.0", "wheel>=0.36.2", "flake8>=3.8.3", "pytest>=6.0.0", "sparsezoo==1.7.0"]}
+    return {"dev": ["black==22.12.0", "isort==5.8.0", "wheel>=0.36.2", "flake8>=3.8.3", "pytest>=6.0.0",]}
 
 setup(
     name="sparsetensors",
diff --git a/src/sparsetensors/quantization/observers/base.py b/src/sparsetensors/quantization/observers/base.py
index a3184096..52a464b9 100644
--- a/src/sparsetensors/quantization/observers/base.py
+++ b/src/sparsetensors/quantization/observers/base.py
@@ -15,7 +15,7 @@
 from typing import Optional, Tuple
 
 from sparsetensors.quantization.quant_args import QuantizationArgs
-from sparsezoo.utils.registry import RegistryMixin
+from sparsetensors.registry.registry import RegistryMixin
 from torch import FloatTensor, IntTensor, Tensor
 from torch.nn import Module