Fix small bug ;)