just small bug fixing