FIX: Fix __init__