Fixes a failing unit test.