Add a couple missing units