Fixes @return doc