Fix the JS