1 files changed, 3 insertions, 1 deletions
diff --git a/os/src/malloc.c b/os/src/malloc.c
index a24a87f..281a75b 100644
--- a/os/src/malloc.c
+++ b/os/src/malloc.c
@@ -1,6 +1,7 @@
 #include <reent.h>
 #include <malloc.h>
-#include <osdebug.h>
+#include <FreeRTOS.h>
+#include <semphr.h>
 
 static xSemaphoreHandle malloc_sem = NULL;
 
@@ -16,4 +17,5 @@ void * malloc(size_t size) {
     ptr =_malloc_r(_impure_ptr, size);
     if (malloc_sem)
         xSemaphoreGive(malloc_sem);
+    return ptr;
 }