Fixed pmf_test.py