1 files changed, 3 insertions, 2 deletions
diff --git a/test/lsan/TestCases/use_stacks_threaded.cc b/test/lsan/TestCases/use_stacks_threaded.cc
index fc4e661c2..a1d4383e9 100644
--- a/test/lsan/TestCases/use_stacks_threaded.cc
+++ b/test/lsan/TestCases/use_stacks_threaded.cc
@@ -7,6 +7,7 @@
 
 #include <assert.h>
 #include <pthread.h>
+#include <sched.h>
 #include <stdio.h>
 #include <stdlib.h>
 
@@ -18,7 +19,7 @@ void *stacks_thread_func(void *arg) {
   fflush(stderr);
   __sync_fetch_and_xor(sync, 1);
   while (true)
-    pthread_yield();
+    sched_yield();
 }
 
 int main() {
@@ -27,7 +28,7 @@ int main() {
   int res = pthread_create(&thread_id, 0, stacks_thread_func, &sync);
   assert(res == 0);
   while (!__sync_fetch_and_xor(&sync, 0))
-    pthread_yield();
+    sched_yield();
   return 0;
 }
 // CHECK: Test alloc: [[ADDR:.*]].