diff --git a/tools/magfit_WMM.py b/tools/magfit_WMM.py
index 1e53891df201be396ea12986b51d28ab35393fd5..06de8926e2dea926dd70c54ff3241ba2469dbf1b 100755
--- a/tools/magfit_WMM.py
+++ b/tools/magfit_WMM.py
@@ -287,7 +287,7 @@ def magfit(logfile):
 
     if args.att_source is not None:
         ATT_NAME = args.att_source
-    if parameters['AHRS_EKF_TYPE'] == 2:
+    elif parameters['AHRS_EKF_TYPE'] == 2:
         ATT_NAME = 'NKF1'
     elif parameters['AHRS_EKF_TYPE'] == 3:
         ATT_NAME = 'XKF1'