diff --git a/bitmind/validator/forward.py b/bitmind/validator/forward.py
index 16fe5ca..4c72b3a 100644
--- a/bitmind/validator/forward.py
+++ b/bitmind/validator/forward.py
@@ -44,6 +44,7 @@ def determine_challenge_type(media_cache, fake_prob=0.5):
         elif modality == 'image':
             # 20% chance to use i2i (in-painting)
             task = 'i2i' if np.random.rand() < 0.2 else 't2i'
+            label = 2
         cache = cache[task]
     return label, modality, task, cache