diff --git a/system76.c b/system76.c
index 9563513..ec076ac 100644
--- a/system76.c
+++ b/system76.c
@@ -170,7 +170,7 @@ static void s76_wmi_notify(u32 value, void *context) {
 	}
 }
 
-static int s76_probe(struct platform_device *dev) {
+static int __init s76_probe(struct platform_device *dev) {
 	int err;
 
 	if (driver_flags & DRIVER_AP_LED) {