fixed a small bug. nice.