summaryrefslogtreecommitdiff
path: root/cmd/disk.c
diff options
context:
space:
mode:
Diffstat (limited to 'cmd/disk.c')
-rw-r--r--cmd/disk.c4
1 files changed, 2 insertions, 2 deletions
diff --git a/cmd/disk.c b/cmd/disk.c
index 3d7bc2f601..92eaa02f4a 100644
--- a/cmd/disk.c
+++ b/cmd/disk.c
@@ -40,8 +40,8 @@ int common_diskboot(struct cmd_tbl *cmdtp, const char *intf, int argc,
bootstage_mark(BOOTSTAGE_ID_IDE_BOOT_DEVICE);
- part = blk_get_device_part_str(intf, (argc == 3) ? argv[2] : NULL,
- &dev_desc, &info, 1);
+ part = blk_get_device_part_str(intf, cmd_arg2(argc, argv),
+ &dev_desc, &info, 1);
if (part < 0) {
bootstage_error(BOOTSTAGE_ID_IDE_TYPE);
return 1;