Remove warning from tests