diff --git a/init/main.c b/init/main.c
index af310af..7cc4b78 100644
--- a/init/main.c
+++ b/init/main.c
@@ -692,7 +692,7 @@
 
 	if (preempt_count() != count) {
 		sprintf(msgbuf, "preemption imbalance ");
-		preempt_count() = count;
+		preempt_count_set(count);
 	}
 	if (irqs_disabled()) {
 		strlcat(msgbuf, "disabled interrupts ", sizeof(msgbuf));