change import in tests