Update include to accommodate name change