fix another small bug