Fix incorrect class header info