Refine feature tests to work with default data.

- Refine feature tests to work with default data.