diff --git a/src/ansys/tools/visualization_interface/__init__.py b/src/ansys/tools/visualization_interface/__init__.py
index 12fcb96e..ca4341b6 100644
--- a/src/ansys/tools/visualization_interface/__init__.py
+++ b/src/ansys/tools/visualization_interface/__init__.py
@@ -20,11 +20,10 @@
 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 # SOFTWARE.
 """Visualization Interface Tool is a Python client library for visualizing the results of Ansys simulations."""
+import importlib.metadata as importlib_metadata
 import os
 
-import pkg_resources
-
-__version__ = pkg_resources.get_distribution("ansys-tools-visualization-interface").version
+__version__ = importlib_metadata.version(__name__.replace(".", "-"))
 
 USE_TRAME: bool = False
 DOCUMENTATION_BUILD: bool = False
diff --git a/src/ansys/tools/visualization_interface/backends/pyvista/pyvista_interface.py b/src/ansys/tools/visualization_interface/backends/pyvista/pyvista_interface.py
index 3e447d0e..0a20caa6 100644
--- a/src/ansys/tools/visualization_interface/backends/pyvista/pyvista_interface.py
+++ b/src/ansys/tools/visualization_interface/backends/pyvista/pyvista_interface.py
@@ -74,7 +74,8 @@ def __init__(
         # Generate custom scene if ``None`` is provided
         if scene is None:
             scene = pv.Plotter(plotter_kwargs)
-
+        if TESTING_MODE:
+            scene.off_screen = True
         # If required, use a white background with no gradient
         if not color_opts:
             color_opts = dict(color="white")