fixed other small bug