Fix @return value