Add missing compilation changes