Fixed small bug.