address comment

PlasmaControl · Aug 29, 2024 · cd2a041 · cd2a041
1 parent 7bb7bd3
commit cd2a041
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/desc/magnetic_fields/_core.py b/desc/magnetic_fields/_core.py
@@ -613,7 +613,7 @@ def save_mgrid(
         )
         bz_001[:] = B_Z
 
-        if A_R is not None:
+        if save_vector_potential:
             ar_001 = file.createVariable("ar_001", np.float64, ("phi", "zee", "rad"))
             ar_001.long_name = (
                 "A_R = radial component of magnetic vector potential "