diff --git a/t_loss/losses.py b/t_loss/losses.py
index 8c6f8f3..4f746d8 100644
--- a/t_loss/losses.py
+++ b/t_loss/losses.py
@@ -1,6 +1,7 @@
-import tensorflow as tf
 import math
 
+import tensorflow as tf
+
 
 class TLoss(tf.keras.losses.Loss):
     """