come on bady
fix a small bug