diff --git a/unit-tests/fml_tests/test_replace_subject.py b/unit-tests/fml_tests/test_replace_subject.py
index 4b1ade0f1..0c15c8624 100644
--- a/unit-tests/fml_tests/test_replace_subject.py
+++ b/unit-tests/fml_tests/test_replace_subject.py
@@ -5,7 +5,7 @@
 from firedrake import (UnitSquareMesh, FunctionSpace, Function, TestFunction,
                        VectorFunctionSpace, MixedFunctionSpace, dx, inner,
                        TrialFunctions, TrialFunction, split)
-from gusto.fml import (Label subject, replace_subject, replace_test_function,
+from gusto.fml import (Label, subject, replace_subject, replace_test_function,
                        replace_trial_function)
 import pytest