Update to make scenario pass