diff --git a/km3buu/tests/test_propagation.py b/km3buu/tests/test_propagation.py
index 87e474ab6aebaffa5a6afa983a8d63a9bb6117e2..27d17044e99561100afe8ee46058e0732a1d1420 100644
--- a/km3buu/tests/test_propagation.py
+++ b/km3buu/tests/test_propagation.py
@@ -14,6 +14,7 @@ import csv
 import unittest
 import numpy as np
 import uproot
+import pytest
 from os.path import abspath, join, dirname
 from thepipe.logger import get_logger