changed test to cover feature logic