Ui test update

Do we need a null check?