Add missing type parameter 'A' to forallM