diff --git a/tests/test_run.py b/tests/test_run.py
index af9b1a8..f72a4ed 100644
--- a/tests/test_run.py
+++ b/tests/test_run.py
@@ -1,3 +1,5 @@
+import pytest
+import numpy as np
 from raking.run_raking import run_raking