diff --git a/target/linux/kirkwood/base-files/lib/upgrade/platform.sh b/target/linux/kirkwood/base-files/lib/upgrade/platform.sh
index be5f1016736770ce2896a3ade0a1dcb193c972e9..e7c6a1120a90147380f4fcf644eb7078c1bc1659 100644
--- a/target/linux/kirkwood/base-files/lib/upgrade/platform.sh
+++ b/target/linux/kirkwood/base-files/lib/upgrade/platform.sh
@@ -14,7 +14,7 @@ platform_check_image() {
 		}
 		return 0
 		;;
-	"dockstar")
+	*)
 		nand_do_platform_check $board $1
 		return $?
 		;;
@@ -32,6 +32,9 @@ platform_do_upgrade() {
 	"linksys-viper")
 		platform_do_upgrade_linksys "$ARGV"
 		;;
+	*)
+		default_do_upgrade "$@"
+		;;
 	esac
 }
 
@@ -39,7 +42,9 @@ platform_pre_upgrade() {
 	local board=$(kirkwood_board_name)
 
 	case "$board" in
-	"dockstar")
+	"linksys-audi"|\
+	"linksys-viper") ;;
+	*)
 		nand_do_upgrade $1
 		;;
 	esac