diff --git a/README.md b/README.md
index 19d63ac..165ebb0 100644
--- a/README.md
+++ b/README.md
@@ -45,7 +45,7 @@ model = Glom(
 )
 
 img = torch.randn(1, 3, 224, 224)
-all_levels = model(img, iters = 12, return_all = True) # (13, 1, 256, 6, 512) - (batch, patches, levels, dimension)
+all_levels = model(img, iters = 12, return_all = True) # (13, 1, 256, 6, 512) - (time, batch, patches, levels, dimension)
 
 # get the top level outputs after iteration 6
 top_level_output = all_levels[7, :, :, -1] # (1, 256, 512) - (batch, patches, dimension)