Use correct import paths

This is now a copy-pastable example.