Fix for a unit test