Fix a test to match comment