update the imports in the demo to use the umbrella header