diff --git a/MemoryModule.c b/MemoryModule.c
index 9f95a70..abbefdd 100644
--- a/MemoryModule.c
+++ b/MemoryModule.c
@@ -54,7 +54,11 @@
 #define IMAGE_SIZEOF_BASE_RELOCATION (sizeof(IMAGE_BASE_RELOCATION))
 #endif
 
-#ifdef _WIN64
+#if defined(_M_ARM64)
+#define HOST_MACHINE IMAGE_FILE_MACHINE_ARM64
+#elif defined(_M_ARM)
+#define HOST_MACHINE IMAGE_FILE_MACHINE_ARM
+#elif defined(_WIN64)
 #define HOST_MACHINE IMAGE_FILE_MACHINE_AMD64
 #else
 #define HOST_MACHINE IMAGE_FILE_MACHINE_I386