Fixes missing #endif