Fixed test with lambda