Minor JS tweak