now fix the JS