diff --git a/eli5/formatters/image.py b/eli5/formatters/image.py
index <HASH>..<HASH> 100644
--- a/eli5/formatters/image.py
+++ b/eli5/formatters/image.py
@@ -46,6 +46,9 @@ def format_as_image(expl,
     image = expl.image
     heatmap = expl.heatmap
     
+    # We first 1. colorize 2. resize
+    # as opposed 1. resize 2. colorize
+
     heatmap = colorize(heatmap, colormap=colormap)
     # TODO: test colorize with a callable
 
diff --git a/eli5/keras.py b/eli5/keras.py
index <HASH>..<HASH> 100644
--- a/eli5/keras.py
+++ b/eli5/keras.py
@@ -103,7 +103,7 @@ def validate_doc(estimator, doc):
     if len(input_sh) == 4:
         # rank 4 with (batch, ...) shape
         # check that we have only one image (batch size 1)
-        single_batch = (1, *input_sh[1:])
+        single_batch = (1, input_sh[1], input_sh[2], input_sh[3])
         if doc_sh != single_batch:
             raise ValueError('Batch size does not match. ' 
                              'doc must be of shape: {}, '
Replace starred expression with manual indexing (py<I> build fix)