diff --git a/doc/update_readme.py b/doc/update_readme.py
index a43b9ef..71bf0ec 100644
--- a/doc/update_readme.py
+++ b/doc/update_readme.py
@@ -29,7 +29,7 @@ def update_readme(readme_path, docstrings):
         file.writelines(new_lines)
 
 if __name__ == "__main__":
-    import plotting
+    import hippunfold_plot.plotting
     readme_path = 'README.md'
     docstrings = get_function_docstrings(plotting)
     update_readme(readme_path, docstrings)
\ No newline at end of file