diff --git a/tests/integration_tests/models/test_reslogit.py b/tests/integration_tests/models/test_reslogit.py
index fa74cd9..1cb0f4e 100644
--- a/tests/integration_tests/models/test_reslogit.py
+++ b/tests/integration_tests/models/test_reslogit.py
@@ -1,8 +1,7 @@
 """Tests ResLogit."""
 
-import pytest
-
 import numpy as np
+import pytest
 import tensorflow as tf
 
 from choice_learn.datasets import load_swissmetro