diff --git a/ufedmm/tests/test_ufedmm.py b/ufedmm/tests/test_ufedmm.py
index fc32263..d76641c 100644
--- a/ufedmm/tests/test_ufedmm.py
+++ b/ufedmm/tests/test_ufedmm.py
@@ -7,12 +7,12 @@
 import sys
 from copy import deepcopy
 
+import numpy as np
 import openmm
 import pytest
 from openmm import unit
 
 import ufedmm
-import numpy as np
 
 
 def ufed_model(