diff --git a/src/tests/test_analysis/test_gaussian_beam/test_fit_beam_data.py b/src/tests/test_analysis/test_gaussian_beam/test_fit_beam_data.py
index 2d93d22e450087db5a782d44c9ba5f30844579bc..c219b036c07e581a23721e01b1436325a31ae13d 100644
--- a/src/tests/test_analysis/test_gaussian_beam/test_fit_beam_data.py
+++ b/src/tests/test_analysis/test_gaussian_beam/test_fit_beam_data.py
@@ -1,10 +1,8 @@
 from pathlib import Path
 from unittest import TestCase
 
-import numpy as np
-
 from openqlab import io
-from openqlab.analysis.gaussian_beam import GaussianBeam, fit_beam_data
+from openqlab.analysis.gaussian_beam import fit_beam_data
 
 filedir = Path(__file__).parent