Fixed UI tests