diff --git a/leakpro/models.py b/leakpro/models.py
index e98f21b..787cfb3 100644
--- a/leakpro/models.py
+++ b/leakpro/models.py
@@ -1,10 +1,10 @@
 """Models for the datasets."""
-from leakpro.import_helper import Self
-
 import torch
 import torch.nn.functional as F  # noqa: N812
 from torch import nn
 
+from leakpro.import_helper import Self
+
 
 class NN(nn.Module):
     """NN for Adult dataset."""