Fixing small bug.