Add missing #endif