Fix generated JS