Fixing a couple errors