Fix small bug with last change.