Fix feature test

- Fix feature test.