diff --git a/febio_python/utils/pyvista_utils.py b/febio_python/utils/pyvista_utils.py
index 335140a..291646b 100644
--- a/febio_python/utils/pyvista_utils.py
+++ b/febio_python/utils/pyvista_utils.py
@@ -2,7 +2,7 @@
 import numpy as np
 from copy import deepcopy
 from pathlib import Path
-from febio_python import FEBioContainer
+from febio_python.container import FEBioContainer
 from febio_python.feb import FebType, Feb25, Feb30, Feb40
 from febio_python.xplt import Xplt
 
@@ -1186,4 +1186,4 @@ def map(cell_type):
         return PyvistaToFEBioElementType.MAPPING.get(cell_type, None)
 
     def __call__(self, cell_type):
-        return self.map(cell_type)
\ No newline at end of file
+        return self.map(cell_type)