diff --git a/ktch/landmark/_plot/_tps.py b/ktch/landmark/_plot/_tps.py
index 778f93b..8e84311 100644
--- a/ktch/landmark/_plot/_tps.py
+++ b/ktch/landmark/_plot/_tps.py
@@ -14,7 +14,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import matplotlib.pyplot as plt
 import numpy as np
 
 from .._Procrustes_analysis import _thin_plate_spline_2d, _tps_2d
@@ -46,6 +45,8 @@ def tps_grid_2d_plot(
         Matplotlib axes.
     """
 
+    import matplotlib.pyplot as plt
+
     W, c, A = _thin_plate_spline_2d(x_reference, x_target)
 
     if ax is None: