diff --git a/tests/requirements.txt b/tests/requirements.txt
index 6fd74ad..df05b4d 100644
--- a/tests/requirements.txt
+++ b/tests/requirements.txt
@@ -1 +1,2 @@
-jaxlib
\ No newline at end of file
+jaxlib
+pytest