diff --git a/Analysis/plot_snapshot.py b/Analysis/plot_snapshot.py
index bdabff5f9a28ec140f233a1f8ccdf91418c14307..b1c1d3736e5fcedc6614b59fed7d108a22d241f6 100755
--- a/Analysis/plot_snapshot.py
+++ b/Analysis/plot_snapshot.py
@@ -33,13 +33,13 @@ def snapshot(k, ax, k1, ishape_last, reader_extr, reader_outer,
 
     dashed: boolean."""
     
+    assert k >= k1
+    
     if k == k1:
         ishape_first = 0
-    elif k > k1:
-        ishape_first = ishape_last[k - k1 - 1] + 1
     else:
-        print("bad k value")
-        return
+        # k > k1
+        ishape_first = ishape_last[k - k1 - 1] + 1
 
     ax.set_title(f"k = {k}", y = 1.05)