Actually, you should mark the box for both because this will indicate you are entering the most up to date form.
Both the IRS and your state tax agency will receive a copy of both W2s from your employer, so it is good to note that you are entering the correct form.
