diff --git a/mojoshader_internal.h b/mojoshader_internal.h
index 0ea43fd..0e53c51 100644
--- a/mojoshader_internal.h
+++ b/mojoshader_internal.h
@@ -16,7 +16,10 @@
 
 /* FIXME: These includes are needed for alloca :( */
 #include <stdlib.h>
-#ifndef __APPLE__
+#if defined(__linux__) || defined(__sun)
+#include <alloca.h>
+#endif
+#ifdef _MSC_VER
 #include <malloc.h>
 #endif
 
@@ -259,10 +262,6 @@ typedef int32_t int32;
 typedef int64_t int64;
 typedef uint64_t uint64;
 #endif
-
-#ifdef sun
-#include <alloca.h>
-#endif
 #endif /* MOJOSHADER_USE_SDL_STDLIB */
 
 #ifdef __GNUC__