Update JS to fix Issue #1