Ui test update

This has been already added in the latest pull request.