diff --git a/pyproject.toml b/pyproject.toml
index ba578bd..2c38fe6 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -19,6 +19,7 @@ requests = "^2.27.1"
 rich = "^12.6.0"
 attrs = "^22.2.0"
 pandas = "^1.1.5"
+numpy = "^1.19.5"
 appdirs = "^1.4.4"
 importlib-metadata = "^4.8.2"
 
diff --git a/tests/integration_tests/conftest.py b/tests/integration_tests/conftest.py
index 3d25a67..5534ab3 100644
--- a/tests/integration_tests/conftest.py
+++ b/tests/integration_tests/conftest.py
@@ -43,7 +43,7 @@
 CONFIGURATIONS_PATH = os.path.join(FILES_PATH, "configuration")
 CONFIGURATION_PATH = os.path.join(CONFIGURATIONS_PATH, "localhost_config.json")
 MODELS_PATH = os.path.join(FILES_PATH, "models")
-MODEL_VERSION = os.environ.get("AERIE_VERSION", "2.8.0")
+MODEL_VERSION = os.environ.get("AERIE_VERSION", "2.11.0")
 MODEL_JAR = os.path.join(MODELS_PATH, f"banananation-{MODEL_VERSION}.jar")
 MODEL_NAME = "banananation"
 MODEL_VERSION = "0.0.1"