Adjust the JS setup