Small correction in __init__