fix potential for a small bug