Bug fix (JS)