Fixed the JS UI