tests: fixed an import which failed the tests.