Minor fix to js file