Fix feature tests

- Fix feature tests.