Add proper return value for edit