Small JS modification