cleaned import in tests