Restore the try check for argument