[KMTEST]

[reactos.git] / rostests / kmtests / ntos_mm / ZwAllocateVirtualMemory.c

diff --git a/rostests/kmtests/ntos_mm/ZwAllocateVirtualMemory.c b/rostests/kmtests/ntos_mm/ZwAllocateVirtualMemory.c
index 67a8ec2..2d7b7b7 100644 (file)
--- a/rostests/kmtests/ntos_mm/ZwAllocateVirtualMemory.c
+++ b/rostests/kmtests/ntos_mm/ZwAllocateVirtualMemory.c
@@ -437,7 +437,13 @@ SystemProcessTest(VOID)
         goto cleanup;
     }
 
-    Status = ObReferenceObjectByHandle(Thread1, THREAD_ALL_ACCESS, PsThreadType, KernelMode, &ThreadObjects[0], NULL);
+    Status = ObReferenceObjectByHandle(Thread1, THREAD_ALL_ACCESS,
+#ifdef _PROPER_NT_EXPORTS
+                                       *PsThreadType,
+#else
+                                       PsThreadType,
+#endif
+                                       KernelMode, &ThreadObjects[0], NULL);
     if (!NT_SUCCESS(Status))
     {
         trace("error referencing thread1\n");
@@ -451,7 +457,13 @@ SystemProcessTest(VOID)
         goto cleanup;
     }
 
-    Status = ObReferenceObjectByHandle(Thread2, THREAD_ALL_ACCESS, PsThreadType, KernelMode, &ThreadObjects[1], NULL);
+    Status = ObReferenceObjectByHandle(Thread2, THREAD_ALL_ACCESS,
+#ifdef _PROPER_NT_EXPORTS
+                                       *PsThreadType,
+#else
+                                       PsThreadType,
+#endif
+                                       KernelMode, &ThreadObjects[1], NULL);
     if (!NT_SUCCESS(Status))
     {
         trace("error referencing thread2\n");