diff --git a/pytorch_finufft/functional.py b/pytorch_finufft/functional.py
index 3aaf62a..3f410ab 100644
--- a/pytorch_finufft/functional.py
+++ b/pytorch_finufft/functional.py
@@ -4,8 +4,8 @@
 
 from typing import Any, Dict, Optional, Tuple, Union
 
-import numpy as np
 import finufft
+import numpy as np
 
 try:
     import cufinufft
diff --git a/tests/test_2d/test_backward_2d.py b/tests/test_2d/test_backward_2d.py
index 751f364..2aba3da 100644
--- a/tests/test_2d/test_backward_2d.py
+++ b/tests/test_2d/test_backward_2d.py
@@ -1,3 +1,5 @@
+from functools import partial
+
 import numpy as np
 import pytest
 import torch
@@ -5,8 +7,6 @@
 
 import pytorch_finufft
 
-from functools import partial
-
 torch.set_default_tensor_type(torch.DoubleTensor)
 torch.set_default_dtype(torch.float64)
 torch.manual_seed(0)