fixed small bug, still bad