fixes return type annotations