Fixed Comperator to WeightedState. again