Fixed the JS