diff --git a/trieste/models/gpflux/models.py b/trieste/models/gpflux/models.py
index fbbfd77a4..b30b14c0c 100644
--- a/trieste/models/gpflux/models.py
+++ b/trieste/models/gpflux/models.py
@@ -19,7 +19,6 @@
 import dill
 import gpflow
 import tensorflow as tf
-from check_shapes import inherit_check_shapes
 from gpflow.inducing_variables import InducingPoints
 from gpflow.keras import tf_keras
 from gpflux.layers import GPLayer, LatentVariableLayer