Fix receiving widget not saving state properly