wta: Allow to restrict number of different monoid values

Otherwise states get one of 2^64 different values in the initial partition,
meaning that they will be distinguished already.