Remove unpicklable lambda