Index: sys/compat/linuxkpi/common/src/linux_schedule.c
===================================================================
--- sys/compat/linuxkpi/common/src/linux_schedule.c
+++ sys/compat/linuxkpi/common/src/linux_schedule.c
@@ -332,11 +332,12 @@
 		sleepq_release(task);
 		ret = 0;
 	}
-	set_task_state(task, TASK_RUNNING);
 
 	if (timeout == 0)
 		return (MAX_SCHEDULE_TIMEOUT);
 
+	set_task_state(task, TASK_RUNNING);
+
 	remainder = compute_timeout_remainder(ret, timeout, remainder);
 
 	return (remainder);