Update @return statements