Little edit in js