unit test for new feature?