Small fix on js file