This fixed the second failing unit test.