Fix a bug in a test