Fix: Minor JS fix