Fix a type mismatch