diff --git a/examples/lookup-static/lookup-static.cpp b/examples/lookup-static/lookup-static.cpp
index 24939a132..237f35e68 100644
--- a/examples/lookup-static/lookup-static.cpp
+++ b/examples/lookup-static/lookup-static.cpp
@@ -243,6 +243,7 @@ int main(int argc, char ** argv){
                 }
 
                 draft.push_back(item_it->second);
+                llama_batch_add(batch_tgt, item_it->second, n_past + i + 1, { 0 }, true);
                 ++n_drafted;
             }
             return;