set right import for tests on Travis