Fix problem with header

Change-Id: I3066952cc95cbc975cbd331ead0fb92238a4e05c