fix import warnings in tests