Ui test update

I suppose these could be externalised?