Fix winner checking function