Options n_states_diag in ezfio corrected