X-Git-Url: https://scm.lunaixsky.com/lunaix-os.git/blobdiff_plain/2687264120b44704961375849a51d69b17ce585a..a2e1b06:/lunaix-os/kernel/ds/semaphore.c

diff --git a/lunaix-os/kernel/ds/semaphore.c b/lunaix-os/kernel/ds/semaphore.c
index c3b1082..b4d0674 100644
--- a/lunaix-os/kernel/ds/semaphore.c
+++ b/lunaix-os/kernel/ds/semaphore.c
@@ -1,5 +1,5 @@
 #include <lunaix/ds/semaphore.h>
-#include <lunaix/sched.h>
+#include <lunaix/kpreempt.h>
 
 void
 sem_init(struct sem_t* sem, unsigned int initial)
@@ -12,7 +12,7 @@ sem_wait(struct sem_t* sem)
 {
     while (!atomic_load(&sem->counter)) {
         // FIXME: better thing like wait queue
-        sched_yieldk();
+        preempt_current();
     }
     atomic_fetch_sub(&sem->counter, 1);
 }