diff --git a/tests/test_prediction.py b/tests/test_prediction.py
index cc56c743afb204a8fafac8930d01c78c4fbb6bf3..b3a1bf578468a23f6977993eb897f6a25c217423 100644
--- a/tests/test_prediction.py
+++ b/tests/test_prediction.py
@@ -3,6 +3,7 @@
 import json
 import shutil
 
+import numpy as np
 import pytest
 import yaml