[cmd] Slight adjust for new api.