Fix failing tests

Remove wrapping object from the selenium elements.