Fixes a unit test