fix incorrect return type