Minor fixes for failing markov model tests