Adds missing @return doc to setInput