Small JavaScript fix