diff --git a/src/main.py b/src/main.py
index 0350c45..e0b8990 100644
--- a/src/main.py
+++ b/src/main.py
@@ -1,6 +1,6 @@
 from argparse import ArgumentParser, Namespace
 
-from model import train_model, infer_sequence
+from model import infer_sequence, train_model
 
 
 def parse_arguments() -> Namespace:
diff --git a/src/model.py b/src/model.py
index b66dcb2..e163c6e 100644
--- a/src/model.py
+++ b/src/model.py
@@ -4,8 +4,8 @@ from numpy import argmax
 from tensorflow import one_hot
 from tensorflow.keras import Model, Sequential
 from tensorflow.keras.layers import Dense, Dropout, Input, Masking
-from tensorflow.keras.models import load_model
 from tensorflow.keras.losses import categorical_crossentropy
+from tensorflow.keras.models import load_model
 from tensorflow.keras.optimizers import Adam
 from tensorflow.keras.regularizers import l2
 from tensorflow.random import set_seed