Fix missing header file