diff --git a/target/linux/brcm47xx/base-files/etc/diag.sh b/target/linux/brcm47xx/base-files/etc/diag.sh
index 3b08554c64..539af36c51 100644
--- a/target/linux/brcm47xx/base-files/etc/diag.sh
+++ b/target/linux/brcm47xx/base-files/etc/diag.sh
@@ -1,28 +1,38 @@
 #!/bin/sh
 # Copyright (C) 2006 OpenWrt.org
 
-set_led() {
-	local led="$1"
-	local state="$2"
-	[ -f "/proc/diag/led/$1" ] && echo "$state" > "/proc/diag/led/$1"
+. /lib/functions/leds.sh
+
+get_status_led() {
+	status_led_file=$(find /sys/class/leds/ -name "*power*" |head -n1)
+	if [ ! -f $status_led_file ]; then
+		status_led=$(basename $status_led_file)
+		return
+	fi;
+	status_led_file=$(find /sys/class/leds/ -name "*diag*" |head -n1)
+	if [ ! -f $status_led_file ]; then
+		status_led=$(basename $status_led_file)
+		return
+	fi;
+	status_led_file=$(find /sys/class/leds/ -name "*wps*" |head -n1)
+	if [ ! -f $status_led_file ]; then
+		status_led=$(basename $status_led_file)
+		return
+	fi;
 }
 
 set_state() {
+	get_status_led
+
 	case "$1" in
-		preinit)
-			set_led dmz 1
-			set_led diag 1
-			set_led power 0
+	preinit)
+		status_led_blink_preinit
 		;;
-		failsafe)
-			set_led diag f
-			set_led power f
-			set_led dmz f
+	failsafe)
+		status_led_blink_failsafe
 		;;
-		done)
-			set_led dmz 0
-			set_led diag 0
-			set_led power 1
+	done)
+		status_led_on
 		;;
 	esac
 }