Fixes failing unit test.