diff --git a/tests/trainers/test_detection.py b/tests/trainers/test_detection.py
index b10e357e2f0..a720c0a6bda 100644
--- a/tests/trainers/test_detection.py
+++ b/tests/trainers/test_detection.py
@@ -14,11 +14,7 @@
 from pytest import MonkeyPatch
 from torch.nn.modules import Module
 
-from torchgeo.datamodules import (
-    MisconfigurationException,
-    NASAMarineDebrisDataModule,
-    VHR10DataModule,
-)
+from torchgeo.datamodules import MisconfigurationException, NASAMarineDebrisDataModule
 from torchgeo.datasets import NASAMarineDebris
 from torchgeo.trainers import ObjectDetectionTask