Fixed minor type error