Trans/Expand - #[main] attribute

2 files changed, 2 insertions, 2 deletions

diff --git a/src/trans/codegen_c.cpp b/src/trans/codegen_c.cpp
index 11a74b4f..7308a9f3 100644
--- a/src/trans/codegen_c.cpp
+++ b/src/trans/codegen_c.cpp
@@ -104,7 +104,7 @@ namespace {
                 m_of
                     << "int main(int argc, const char* argv[]) {\n"
                     << "\t" << Trans_Mangle( ::HIR::GenericPath(m_resolve.m_crate.get_lang_item_path(Span(), "start")) ) << "("
-                        << "(uint8_t*)" << Trans_Mangle( ::HIR::GenericPath(::HIR::SimplePath(m_crate.m_crate_name, {"main"})) ) << ", argc, (uint8_t**)argv"
+                        << "(uint8_t*)" << Trans_Mangle( ::HIR::GenericPath(m_resolve.m_crate.get_lang_item_path(Span(), "mrustc-main")) ) << ", argc, (uint8_t**)argv"
                         << ");\n"
                     << "}\n";
             }
diff --git a/src/trans/enumerate.cpp b/src/trans/enumerate.cpp
index 593cd6e1..f3306c65 100644
--- a/src/trans/enumerate.cpp
+++ b/src/trans/enumerate.cpp
@@ -70,7 +70,7 @@ TransList Trans_Enumerate_Main(const ::HIR::Crate& crate)
 
     // user entrypoint
     {
-        auto main_path = ::HIR::SimplePath(crate.m_crate_name, {"main"});
+        auto main_path = crate.get_lang_item_path(Span(), "mrustc-main");
         const auto& fcn = crate.get_function_by_path(sp, main_path);
 
         state.enum_fcn( main_path, fcn, {} );