Fix a test bug