Correct JS code