fixes, small bug