Minor fix to apply mag field