[INTEGRATION TESTS] Small improvement