Replace incorrect return type