Fix lambda usage in the documentation