This fixed one of the failing unit tests.