Fixed: TypeError: load() takes 2 positional arguments but 3 were given; found by @ly0