Fix small JS error