Another small change to fix a small bug.