Added missing #endif

Whoops.