Small change in the JS