small JS fix