diff --git a/relevant_docs/count_relevant_docs.py b/relevant_docs/count_relevant_docs.py
index eace629..072ad89 100644
--- a/relevant_docs/count_relevant_docs.py
+++ b/relevant_docs/count_relevant_docs.py
@@ -5,6 +5,7 @@
 import argparse
 import os
 import utils
+import numpy as np
 
 from country_entities import countries