Added round restoration to testMain2