Fixed ProbabilityModel

I misunderstood how for..in worked.