Fixed a small bug on the waterfall