Fixed a type mismatch