Oops. Another small bug