Update selenium tests for new layout