Fix TypeError: <lambda>() takes no arguments (1 given)