diff --git a/include/tvm/script/printer/traced_object_functor.h b/include/tvm/script/printer/traced_object_functor.h
index 7783801f9ccee..d10a2d6e649dc 100644
--- a/include/tvm/script/printer/traced_object_functor.h
+++ b/include/tvm/script/printer/traced_object_functor.h
@@ -16,15 +16,17 @@
  * specific language governing permissions and limitations
  * under the License.
  */
-#ifndef TVM_SCRIPT_PRINTER_REGISTRY_H_
-#define TVM_SCRIPT_PRINTER_REGISTRY_H_
+#ifndef TVM_SCRIPT_PRINTER_TRACED_OBJECT_FUNCTOR_H_
+#define TVM_SCRIPT_PRINTER_TRACED_OBJECT_FUNCTOR_H_
 
 #include <tvm/node/node.h>
 #include <tvm/runtime/logging.h>
 #include <tvm/runtime/packed_func.h>
 #include <tvm/script/printer/traced_object.h>
 
+#include <string>
 #include <type_traits>
+#include <unordered_map>
 #include <utility>
 #include <vector>
 
@@ -163,4 +165,4 @@ class TracedObjectFunctor {
 }  // namespace printer
 }  // namespace script
 }  // namespace tvm
-#endif  // TVM_SCRIPT_PRINTER_REGISTRY_H_
+#endif  // TVM_SCRIPT_PRINTER_TRACED_OBJECT_FUNCTOR_H_
diff --git a/src/script/printer/traced_object_functor.cc b/src/script/printer/traced_object_functor.cc
index 0314ba8dd3276..64483602a83c5 100644
--- a/src/script/printer/traced_object_functor.cc
+++ b/src/script/printer/traced_object_functor.cc
@@ -40,7 +40,7 @@ const runtime::PackedFunc* GetDispatchFunctionForToken(const DispatchTable& tabl
   } else {
     return nullptr;
   }
-};
+}
 
 const runtime::PackedFunc& GetDispatchFunction(const DispatchTable& dispatch_table,
                                                const String& token, uint32_t type_index) {