fix small bug ;)