Update to use the proper include file