Added missing unit test for feature