Fix broken UX tests

Card: http://catchup.io/boards/3/cards/71