small return type fix