Fix the local test
Fix the test.