diff --git a/src/spikeinterface/generation/drift_tools.py b/src/spikeinterface/generation/drift_tools.py
index d65717861e..f76ab71803 100644
--- a/src/spikeinterface/generation/drift_tools.py
+++ b/src/spikeinterface/generation/drift_tools.py
@@ -6,8 +6,7 @@
 import numpy as np
 from numpy.typing import ArrayLike
 from probeinterface import Probe
-from spikeinterface.core import (BaseRecording, BaseRecordingSegment,
-                                 BaseSorting, Templates)
+from spikeinterface.core import BaseRecording, BaseRecordingSegment, BaseSorting, Templates
 
 
 def interpolate_templates(templates_array, source_locations, dest_locations, interpolation_method="cubic"):