diff --git a/src/mednet/libs/classification/config/models/cnn3d.py b/src/mednet/libs/classification/config/models/cnn3d.py
index 305b545287077a6b67c54da4fcca25025a2c6615..e0559d7de03d81d1f858a04dee5ca837f34c536a 100644
--- a/src/mednet/libs/classification/config/models/cnn3d.py
+++ b/src/mednet/libs/classification/config/models/cnn3d.py
@@ -3,11 +3,10 @@
 # SPDX-License-Identifier: GPL-3.0-or-later
 """Simple CNN for 3D organ classification, to be trained from scratch."""
 
+from mednet.models.cnn3d import Conv3DNet
 from torch.nn import BCEWithLogitsLoss
 from torch.optim import Adam
 
-from mednet.models.cnn3d import Conv3DNet
-
 model = Conv3DNet(
     loss_type=BCEWithLogitsLoss,
     optimizer_type=Adam,