diff --git a/test/test_mixed_function_space_with_mixed_mesh.py b/test/test_mixed_function_space_with_mixed_mesh.py
index f336fb903..7a1f6aaa6 100644
--- a/test/test_mixed_function_space_with_mixed_mesh.py
+++ b/test/test_mixed_function_space_with_mixed_mesh.py
@@ -1,3 +1,4 @@
+import pytest
 from ufl import (
     CellVolume,
     Coefficient,
@@ -10,6 +11,7 @@
     SpatialCoordinate,
     TestFunction,
     TrialFunction,
+    div,
     grad,
     inner,
     split,