fix one small bug that makes things work much better