Auto-update imports on Jupyter Notebooks after changing module directory

Refactoring Jupyter Notebook isn't as straightforward as python scripts. Notebooks have to be converted to scripts first before refactoring. Using something like https://nbconvert.readthedocs.io/en/latest/ or https://github.com/mwouts/jupytext.

However, it should be easy to automatically update imports in the notebooks after moving the python module to a different directory? 

I tried to do that in Pycham but it never updates the references in the notebooks. Is there any way to auto-update reference after moving modules to different directories? Is there any configurations to include notebooks in the search for references and update them automatically?