Small update to header file.