Fixing up a couple errors