diff --git a/target/linux/x86/rdc/base-files/etc/diag.sh b/target/linux/x86/rdc/base-files/etc/diag.sh
index 8d5c38262c..03796d7784 100644
--- a/target/linux/x86/rdc/base-files/etc/diag.sh
+++ b/target/linux/x86/rdc/base-files/etc/diag.sh
@@ -1,19 +1,17 @@
 #!/bin/sh
-# Copyright (C) 2007 OpenWrt.org
+# Copyright (C) 2007-2013 OpenWrt.org
 
-set_led() {
-	local led="$1"
-	local state="$2"
-	[ -d "/sys/class/leds/rdc321x:$led" ] && echo "$state" > "/sys/class/leds/rdc321x:$led/brightness"
-}
+. /lib/functions/leds.sh
+
+status_led="rdc321x:dmz"
 
 set_state() {
 	case "$1" in
-		preinit)
-			set_led dmz 1
+	preinit)
+		status_led_on
 		;;
-		done)
-			set_led dmz 0
+	done)
+		status_led_off
 		;;
 	esac
 }