fixes return type annotation