Fix js for new elements