diff --git a/pystencils_tests/test_graph_datahandling.py b/pystencils_tests/test_graph_datahandling.py
index 441b9ff38f4cb4d28a5baee5302f48fc96b68385..ccb0ebf914580cbfadcf5ea3c900ca956fb06b44 100644
--- a/pystencils_tests/test_graph_datahandling.py
+++ b/pystencils_tests/test_graph_datahandling.py
@@ -10,6 +10,8 @@
 
 import pytest
 
+pytest.importorskip('lbmpy')
+
 from lbmpy.boundaries import UBB, FixedDensity, NoSlip
 from lbmpy.geometry import add_pipe_inflow_boundary, add_pipe_walls
 from lbmpy.lbstep import LatticeBoltzmannStep
@@ -17,7 +19,6 @@ from pystencils.datahandling import create_data_handling
 from pystencils.datahandling.graph_datahandling import GraphDataHandling
 from pystencils.slicing import slice_from_direction
 
-pytest.importorskip('lbmpy')
 
 
 def create_lid_driven_cavity(domain_size=None, lid_velocity=0.005, lbm_kernel=None,