Small Javascript tweak