Fix error handling in dx add stage