Fix plotting after refactor

Fix an import line in `plotting.py` that was broken by the recent creation of a `dist` directory.