Only remove <command>'s <arg> elements when using --main