Method return type fix