Small fix for XML generation