Fix for a unit test failure