I played it again and that really does smooth out the feel a lot! Having it only push you when you're in the air is a really nice way to balance it with its usability for putting out the fires!
