Add missing inclusion in extension