Fixing a couple compile errors