Fixing error upon winning