diff --git a/setup.py b/setup.py
index fe350d5..1d264bc 100644
--- a/setup.py
+++ b/setup.py
@@ -39,8 +39,9 @@
     description='Flood-Filling Networks for volumetric instance segmentation',
     long_description=open('README.md').read(),
     install_requires=[
-        'connectomics',
+        'connectomics @ git+https://github.com/google-research/connectomics@main',
         'edt>=2.3.0',
+        'clu',
         'pandas',
         'dataclasses-json>=0.5.6',
         'scikit-image>=0.11.0',