Fix errors from last commit, d'oh