Fix a test

Fix a test