Fix minor type error