Added missing unit-test