Added missing unit test