Minor include fix