fix a small bug in step one