Add correct init