Fixed an issue in a test