Fix import reference