Fix: TypeError: __init__() takes exactly 1 argument (2 given)