Fix implicit return on lambda statements