- fix small bug