super small bug fix