Fix small bug issue.