Fix class name in __init__