Fixes some small errors with the new additions.