diff --git a/.github/workflows/python-package.yml b/.github/workflows/python-package.yml
index 1193ee6b..ecc4bbc0 100644
--- a/.github/workflows/python-package.yml
+++ b/.github/workflows/python-package.yml
@@ -25,6 +25,7 @@ jobs:
     
     - name: Install dependencies
       run: |
+        sudo apt-get install libhdf5-dev
         pip install -U setuptools>=40.1 wheel 'cmake<3.20'
         pip install .
     
diff --git a/setup.py b/setup.py
index 84ed0b75..b9110cd5 100644
--- a/setup.py
+++ b/setup.py
@@ -47,7 +47,7 @@
         'loompy',
         'MulticoreTSNE',
         'Click<8',
-        'umap-learn<0.4.0',
+        'umap-learn',
         'harmonypy>=0.0.5',
         'bbknn>=1.5.0',
         'mnnpy>=0.1.9.5',