diff --git a/tests/test_denoise.py b/tests/test_denoise.py
index ce4c9a4b..ad2110a2 100644
--- a/tests/test_denoise.py
+++ b/tests/test_denoise.py
@@ -7,7 +7,7 @@
 """
 
 import numpy as np
-
+from lasy.profiles.transverse.super_gaussian_profile import SuperGaussianTransverseProfile
 from lasy.utils.denoise import hg_reconstruction