Remove lambda case extension from tests