Fix minor type mismatch