Fix arg vec issue detected by final test