fix a small bug (?)