fix ImportError at tests