diff --git a/alphaviz/thermo/xic_plot.py b/alphaviz/thermo/xic_plot.py
index 44da1b6..bb56373 100644
--- a/alphaviz/thermo/xic_plot.py
+++ b/alphaviz/thermo/xic_plot.py
@@ -11,7 +11,7 @@
TimsTOF,
)
-from alphaviz.contrib.msplot_utils import (
+from alphaviz.thermo.msplot_utils import (
_plot_line, _plot_line_fast
)
diff --git a/nbs/tutorial_DIANN_alphabase.ipynb b/nbs/tutorial_DIANN_alphabase.ipynb
index 966b9df..df1da8a 100644
--- a/nbs/tutorial_DIANN_alphabase.ipynb
+++ b/nbs/tutorial_DIANN_alphabase.ipynb
@@ -36,7 +36,7 @@
}
],
"source": [
- "from alphaviz.contrib.ms_viz import MS_Viz\n",
+ "from alphaviz.thermo.ms_viz import MS_Viz\n",
"\n",
"from peptdeep.pretrained_models import ModelManager\n",
"\n",
@@ -353623,7 +353623,9 @@
"outputs": [
{
"data": {
- "image/svg+xml": ""
+ "image/svg+xml": [
+ ""
+ ]
},
"metadata": {},
"output_type": "display_data"
@@ -353649,7 +353651,9 @@
"outputs": [
{
"data": {
- "image/svg+xml": ""
+ "image/svg+xml": [
+ ""
+ ]
},
"metadata": {},
"output_type": "display_data"