Update import paths

The import paths seem to be out of date for running the example.