diff --git a/lbmpy_tests/test_oldroydb.py b/lbmpy_tests/test_oldroydb.py
index d25054daadc590564b774f87e57a0f6c7ec1d7bf..f7f80fb90ef53739a422bb230f15707d64014429 100755
--- a/lbmpy_tests/test_oldroydb.py
+++ b/lbmpy_tests/test_oldroydb.py
@@ -10,6 +10,7 @@ from lbmpy.boundaries import NoSlip
 import scipy.optimize
 
 from lbmpy.oldroydb import *
+import pytest
 
 
 # # Lattice Boltzmann with Finite-Volume Oldroyd-B