Fix broken JS