Fixed import errors inside one of the tests.