Fixed compilation errors

Fixed compilation errors