small bug and comp fix