small JS change