diff --git a/CI/integration_tests/training_recording/test_loss_ntk_recording_deployment.py b/CI/integration_tests/training_recording/test_loss_ntk_recording_deployment.py
index 33a4c60..d0032bb 100644
--- a/CI/integration_tests/training_recording/test_loss_ntk_recording_deployment.py
+++ b/CI/integration_tests/training_recording/test_loss_ntk_recording_deployment.py
@@ -30,9 +30,9 @@
 os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
 
 import numpy as np
-from numpy.testing import assert_array_almost_equal
 import optax
 from neural_tangents import stax
+from numpy.testing import assert_array_almost_equal
 
 from znnl.loss_functions import LPNormLoss
 from znnl.models import NTModel