* Make sure the tests work accordingly again

* Linting

* Do not use the trait due to issues on previous versions

* Method also does not always exist

* Apply linting