Fix compilation unit extension parsing.