diff --git a/openmc_plotter/plotmodel.py b/openmc_plotter/plotmodel.py
index 11e2e19..d102793 100644
--- a/openmc_plotter/plotmodel.py
+++ b/openmc_plotter/plotmodel.py
@@ -224,7 +224,7 @@ def makePlot(self):
         domain[_NOT_FOUND] = DomainView(_NOT_FOUND, "Not Found", cv.domainBackground)
         u, inv = np.unique(self.ids, return_inverse=True)
         image = np.array([domain[id].color for id in u])[inv]
-        image.shape = (cv.v_res, cv.h_res, 3)
+        image = image.reshape(cv.v_res, cv.h_res, 3)
 
         if cv.masking:
             for id, dom in domain.items():