diff --git a/pyproject.toml b/pyproject.toml
index af2b700..6b2508a 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -52,7 +52,8 @@ dependencies = [
 [project.optional-dependencies]
 train = [
     # Dependencies to train the original OpenPredict drug-disease model
-    "numpy ==1.16.1", # requires python 3.8
+    #"numpy ==1.16.1", # requires python 3.8
+    "numpy ==1.19.5", # requires python 3.8
     "pandas ==1.1.1",
     "torch ==1.13.1",
     "networkx",