Small tweak Javascript