Small change to use the same JS.