Header file fix