Fixes up some JS breaking