Fix highlight in molecules