fixing small bug line 107