tests: modify importing