update to expect new arg format