diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index a6514ca58..b966d0d4a 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -25,6 +25,7 @@ jobs:
         run: |
           sudo apt-get update
           sudo apt-get install -y build-essential libeigen3-dev libfftw3-dev libhdf5-dev libopenmpi-dev
+          sudo pip install numpy
 
       - name: Setup submodule and build
         run: |
diff --git a/tests/Disk/cyl_basis.py b/tests/Disk/cyl_basis.py
index 227b22f4a..c3c55b4e4 100644
--- a/tests/Disk/cyl_basis.py
+++ b/tests/Disk/cyl_basis.py
@@ -1,7 +1,6 @@
 #!/usr/bin/env python
 # coding: utf-8
 
-import numpy as np
 import pyEXP
 
 # Make the disk basis config
diff --git a/tests/Halo/sph_basis.py b/tests/Halo/sph_basis.py
index 1cee5d8ac..f7b6a18b9 100644
--- a/tests/Halo/sph_basis.py
+++ b/tests/Halo/sph_basis.py
@@ -1,7 +1,6 @@
 #!/usr/bin/env python
 # coding: utf-8
 
-import numpy as np
 import pyEXP
 
 # Make the halo basis config