Fix bug in __init__