Fix a test error