diff --git a/code/CMakeModules/FindWindowsSDK.cmake b/code/CMakeModules/FindWindowsSDK.cmake
index fd32d92b5..036e730d5 100644
--- a/code/CMakeModules/FindWindowsSDK.cmake
+++ b/code/CMakeModules/FindWindowsSDK.cmake
@@ -71,11 +71,14 @@ SET(WINSDKENV_DIR $ENV{WINSDK_DIR})
 MACRO(FIND_WINSDK_VERSION_HEADERS)
   IF(WINSDK_DIR AND NOT WINSDK_VERSION)
     # Search version in headers
-    IF(EXISTS ${WINSDK_DIR}/include/Msi.h)
-      SET(_MSI_FILE ${WINSDK_DIR}/include/Msi.h)
-    ENDIF(EXISTS ${WINSDK_DIR}/include/Msi.h)
+    FIND_FILE(_MSI_FILE Msi.h
+      PATHS
+      ${WINSDK_DIR}/Include/um
+      ${WINSDK_DIR}/Include
+    )
 
     IF(_MSI_FILE)
+
       # Look for Windows SDK 8.0
       FILE(STRINGS ${_MSI_FILE} _CONTENT REGEX "^#ifndef NTDDI_WIN8")
 
@@ -88,11 +91,11 @@ MACRO(FIND_WINSDK_VERSION_HEADERS)
         FILE(STRINGS ${_MSI_FILE} _CONTENT REGEX "^#ifndef NTDDI_WIN7")
         
         IF(_CONTENT)
-          IF(EXISTS ${WINSDK_DIR}/include/winsdkver.h)
-            SET(_WINSDKVER_FILE ${WINSDK_DIR}/include/winsdkver.h)
-          ELSEIF(EXISTS ${WINSDK_DIR}/include/WinSDKVer.h)
-            SET(_WINSDKVER_FILE ${WINSDK_DIR}/include/WinSDKVer.h)
-          ENDIF(EXISTS ${WINSDK_DIR}/include/winsdkver.h)
+          FIND_FILE(_WINSDKVER_FILE winsdkver.h WinSDKVer.h
+            PATHS
+            ${WINSDK_DIR}/Include/um
+            ${WINSDK_DIR}/Include
+          )
 
           IF(_WINSDKVER_FILE)
             # Load WinSDKVer.h content
@@ -173,7 +176,7 @@ MACRO(USE_CURRENT_WINSDK)
 
       # Look for Windows.h because there are several paths
       IF(EXISTS ${_INCLUDE}/Windows.h)
-        STRING(REGEX REPLACE "/(include|INCLUDE|Include)" "" WINSDK_DIR ${_INCLUDE})
+        STRING(REGEX REPLACE "/(include|INCLUDE|Include)(.*)" "" WINSDK_DIR ${_INCLUDE})
         MESSAGE(STATUS "Found Windows SDK environment variable in ${WINSDK_DIR}")
         BREAK()
       ENDIF(EXISTS ${_INCLUDE}/Windows.h)