final small bug

final small bug