Remove reference to stage