Fixed logic

Waterpicker <Waterpicker@rocketmail.com>