diff --git a/pytorch_finufft/_err.py b/pytorch_finufft/_err.py
index 26924bc..bb6de3e 100644
--- a/pytorch_finufft/_err.py
+++ b/pytorch_finufft/_err.py
@@ -1,4 +1,4 @@
-from typing import Union, Tuple
+from typing import Tuple, Union
 
 import torch
 
diff --git a/pytorch_finufft/functional.py b/pytorch_finufft/functional.py
index f64a61b..96551b5 100644
--- a/pytorch_finufft/functional.py
+++ b/pytorch_finufft/functional.py
@@ -2,7 +2,7 @@
 Implementations of the corresponding Autograd functions
 """
 
-from typing import Any, Optional, Union, Tuple, Dict
+from typing import Any, Dict, Optional, Tuple, Union
 
 import finufft
 import torch