diff --git a/src/Runtime/CMakeLists.txt b/src/Runtime/CMakeLists.txt
index a5f2a55..0e7742c 100644
--- a/src/Runtime/CMakeLists.txt
+++ b/src/Runtime/CMakeLists.txt
@@ -35,5 +35,5 @@ target_include_directories(PyRuntime PRIVATE
         ${ONNX_MLIR_SRC_ROOT})
 
 add_dependencies(PyRuntime cruntime)
-install(FILES Runtime/DynMemRef.h DESTINATION include)
+install(FILES DynMemRef.h DESTINATION include)
 install(TARGETS cruntime DESTINATION lib)
diff --git a/utils/debug.py b/utils/debug.py
index d8d987d..d966ac9 100644
--- a/utils/debug.py
+++ b/utils/debug.py
@@ -22,15 +22,15 @@ if (not os.environ.get('ONNX_MLIR_HOME', None)):
 VERBOSE = os.environ.get('VERBOSE', False)
 ONNX_MLIR = os.path.join(os.environ['ONNX_MLIR_HOME'], "bin/onnx-mlir")
 
-# Include runtime directory in python paths, so pyruntime can be imported.
+# Include runtime directory in python paths, so PyRuntime can be imported.
 RUNTIME_DIR = os.path.join(os.environ['ONNX_MLIR_HOME'], "lib")
 sys.path.append(RUNTIME_DIR)
 
 try:
-    from pyruntime import ExecutionSession
+    from PyRuntime import ExecutionSession
 except ImportError:
     raise ImportError(
-        "Looks like you did not build the pyruntime target, build it by running `make pyruntime`."
+        "Looks like you did not build the PyRuntime target, build it by running `make PyRuntime`."
     )