Fixed small bug..