Simple fix for the JS.