Tests no longer look for '.link-right.restart'

- Previously our tests were looking for a `<th>` with `'.link-right.restart'`. These classes have been removed and so now we can just inspect the done questions section of the markup for the presence of the link
- Fixes test errors