Added header file missed from previous commit