Fix errors from additions