Fixed changes to header file breaking build