diff --git a/docs/imviz/import_data.rst b/docs/imviz/import_data.rst
index 4693b23eb1..6e94873b23 100644
--- a/docs/imviz/import_data.rst
+++ b/docs/imviz/import_data.rst
@@ -94,8 +94,7 @@ object, you can load it into Imviz as follows:
 
 .. code-block:: python
 
-    import numpy as np
-    from astropy.nddata import NDData
+    from astropy.nddata import NDData, StdDevUncertainty
     from jdaviz import Imviz
 
     # mydatamodel is a jwst.datamodels object with stddev ERR array