Fix import in init