Fixing up a failing unit test.