Fix flight
Fixed #232 - improved potion handling