Small fix to JS