diff --git a/tests/test_methods_cna.py b/tests/test_methods_cna.py
index 448d873..c5ad3e1 100644
--- a/tests/test_methods_cna.py
+++ b/tests/test_methods_cna.py
@@ -1,3 +1,4 @@
+import pytest
 import numpy as np
 
 from oor_benchmark.api import check_method, sample_dataset