From 8c7d99b7d6c6f467a00731d34730b9fbc43981ac Mon Sep 17 00:00:00 2001
From: Jared Roesch <jroesch@octoml.ai>
Date: Mon, 8 Jun 2020 17:45:34 -0700
Subject: [PATCH] Replace with helper

---
 rust/macros/src/external.rs | 7 +------
 1 file changed, 1 insertion(+), 6 deletions(-)

diff --git a/rust/macros/src/external.rs b/rust/macros/src/external.rs
index 06f39f1d2ec3d..a739d01f26503 100644
--- a/rust/macros/src/external.rs
+++ b/rust/macros/src/external.rs
@@ -18,7 +18,6 @@
  */
 use proc_macro2::Span;
 use quote::quote;
-use std::env;
 use syn::parse::{Parse, ParseStream, Result};
 
 use syn::{FnArg, Generics, Ident, Lit, Meta, NestedMeta, Pat, ReturnType, TraitItemMethod, Type};
@@ -87,11 +86,7 @@ impl Parse for ExternalInput {
 pub fn macro_impl(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
     let ext_input = syn::parse_macro_input!(input as ExternalInput);
 
-    let tvm_rt_crate = if env::var("CARGO_PKG_NAME").unwrap() == "tvm-rt" {
-        quote!(crate)
-    } else {
-        quote!(tvm_rt)
-    };
+    let tvm_rt_crate = crate::util::get_tvm_rt_crate();
 
     let err_type = quote! { #tvm_rt_crate::Error };