Fix MAGMOM tests