Fix CI: Rewrite tests which work locally
Fix CI: new failures - element not found