diff --git a/solarmach/tests/test.py b/solarmach/tests/test.py
index 528397f..bbd61b6 100644
--- a/solarmach/tests/test.py
+++ b/solarmach/tests/test.py
@@ -7,6 +7,7 @@
 import numpy as np
 import pandas
 import pfsspy
+import sunpy
 from solarmach import SolarMACH, print_body_list, get_gong_map, calculate_pfss_solution, sc_distance