1 files changed, 3 insertions, 2 deletions
diff --git a/lib/transform-dis.h b/lib/transform-dis.h
index 70fe57a..c1de937 100644
--- a/lib/transform-dis.h
+++ b/lib/transform-dis.h
@@ -1,10 +1,11 @@
 #pragma once
 #include <stdint.h>
 #include <stdbool.h>
+#include stringify(TARGET_DIR/arch-dis.h)
 
 int transform_dis_main(const void *restrict code_ptr,
                        void **restrict rewritten_ptr_ptr,
-                       uintptr_t pc_patch_start,
-                       uintptr_t *pc_patch_end_p,
+                       uint_tptr pc_patch_start,
+                       uint_tptr *pc_patch_end_p,
                        struct arch_dis_ctx *arch_ctx_p,
                        int *offset_by_pcdiff);