fix Integration tests