edit return type