Fix JS

Trying to fix the JS, hope this works