Fixing little compile problems