Fix __init__ stuff