Index: kernel/generic/src/proc/task.c
===================================================================
--- kernel/generic/src/proc/task.c	(revision 07d4271f2d0447f764d751ed8f95104f720791d2)
+++ kernel/generic/src/proc/task.c	(revision e4c0e0c6723e6cff415f2677f4aa49176b1c2534)
@@ -1,5 +1,5 @@
 /*
+ * Copyright (c) 2025 Jiri Svoboda
  * Copyright (c) 2010 Jakub Jermar
- * Copyright (c) 2018 Jiri Svoboda
  * All rights reserved.
  *
@@ -104,5 +104,5 @@
  *
  */
-void task_done(void)
+void task_done(task_t *cur_task)
 {
 	size_t tasks_left;
@@ -112,4 +112,5 @@
 		task_t *task_0 = ipc_box_0->task;
 		ipc_box_0 = NULL;
+
 		/*
 		 * The first task is held by kinit(), we need to release it or
@@ -129,5 +130,5 @@
 		task = task_first();
 		while (task != NULL) {
-			if (task != TASK) {
+			if (task != cur_task) {
 				tasks_left++;
 #ifdef CONFIG_DEBUG