fixed another small bug