diff --git a/tests/test_convolution_fwt.py b/tests/test_convolution_fwt.py
index df51ea47..fa73eaaa 100644
--- a/tests/test_convolution_fwt.py
+++ b/tests/test_convolution_fwt.py
@@ -313,6 +313,9 @@ def _compare_coeffs(
 
     Returns:
         A list with bools from allclose.
+
+    Raises:
+        TypeError: In case of a problem with the list structures.
     """
     test_list: List[bool] = []
     for ptwtcs, pywtcs in zip(ptwt_res, pywt_res):
diff --git a/tests/test_cwt.py b/tests/test_cwt.py
index ff967e83..0008363a 100644
--- a/tests/test_cwt.py
+++ b/tests/test_cwt.py
@@ -1,6 +1,6 @@
 """Test the continuous transformation code."""
 
-from typing import Any, Union
+from typing import Any
 
 import numpy as np
 import pytest
diff --git a/tests/test_jit.py b/tests/test_jit.py
index 93b15372..a7a0d6a8 100644
--- a/tests/test_jit.py
+++ b/tests/test_jit.py
@@ -1,6 +1,6 @@
 """Ensure pytorch's torch.jit.trace feature works properly."""
 
-from typing import Any, Dict, List, NamedTuple, Optional, Tuple, Union
+from typing import Dict, List, NamedTuple, Optional, Tuple, Union
 
 import numpy as np
 import pytest