Fix small bug in run code.