Trying to fix the tests [skip-ci]