This was in the test file previously. It is right above the test I added for probability_transition.