small bug
clean up