fixed bug when adding the new custom element code.