Fix issue with test file