Header file fix (hopefully)