return type fix