small fix for lambda case