This is a temporary fix as using css classes to target
elements in tests is flaky. We will need to find a proper
solution once we have fix this problem.