Fix feature tests.

- Fix feature tests to work with standard set of fake.