Update to match main JS file