Fixed bugs in unit test prob_test.cpp