Make it work with import xml files