Bug fix (lack of header file inclusion)