some cleanup
fixing a small bug