Small unit test fix