removed dependency from tests
fix main package import