Small js error fix