diff --git a/src/dpi/module.cc b/src/dpi/module.cc
index def13054..b39e69ae 100644
--- a/src/dpi/module.cc
+++ b/src/dpi/module.cc
@@ -282,7 +282,7 @@ class DPIModule final : public DPIModuleNode {
   }
 
   PackedFunc GetFunction(
-      const std::string& name,
+      const String& name,
       const ObjectPtr<Object>& sptr_to_self) final {
     if (name == "WriteReg") {
       return TypedPackedFunc<void(int, int)>(