Small JS fix