diff --git a/common/usb_kbd.c b/common/usb_kbd.c
index 2472d25..960a70a 100644
--- a/common/usb_kbd.c
+++ b/common/usb_kbd.c
@@ -487,6 +487,11 @@
 		if (error)
 			return error;
 
+#ifdef CONFIG_CONSOLE_MUX
+		error = iomux_doenv(stdin, stdinname);
+		if (error)
+			return error;
+#else
 		/* Check if this is the standard input device. */
 		if (strcmp(stdinname, DEVNAME))
 			return 1;
@@ -498,6 +503,7 @@
 		error = console_assign(stdin, DEVNAME);
 		if (error)
 			return error;
+#endif
 
 		return 1;
 	}