Fixed an issue with a unit test