Fix class header