fixing JS due to code change