Minor JS fix