diff --git a/solarmach/tests/test.py b/solarmach/tests/test.py
index 81db598..528397f 100644
--- a/solarmach/tests/test.py
+++ b/solarmach/tests/test.py
@@ -86,7 +86,7 @@ def test_solarmach_pfss():
     vsw_list = [400, 400]   # position-sensitive solar wind speed per body in body_list
     sm = SolarMACH(date, body_list, vsw_list, reference_long=100, reference_lat=10)
     gong_map = get_gong_map(time=date, filepath=None)
-    assert isinstance(gong_map, pfsspy.map.GongSynopticMap)
+    assert isinstance(gong_map, pfsspy.map.GongSynopticMap) or isinstance(gong_map, sunpy.map.sources.gong.GONGSynopticMap)
     pfss_solution = calculate_pfss_solution(gong_map=gong_map, rss=2.5, coord_sys='Carrington')
     assert isinstance(pfss_solution, pfsspy.output.Output)
     fig, ax = sm.plot_pfss(rss=2.5, pfss_solution=pfss_solution, vary=True, return_plot_object=True,