diff --git a/main.cpp b/main.cpp
index b64e9d13f..bd35270e0 100644
--- a/main.cpp
+++ b/main.cpp
@@ -250,9 +250,9 @@ int main(int argc, char ** argv) {
     }
 
     // tokenize the first reverse prompt
-    auto first_antiprompt = ::llama_tokenize(ctx,"",false);
+    auto first_antiprompt = ::llama_tokenize(ctx, params.prompt, true);
     if (!params.antiprompt.empty()) {
-       auto first_antiprompt = ::llama_tokenize(ctx, params.antiprompt.front(), false);
+       auto first_antiprompt = ::llama_tokenize(ctx, params.antiprompt.front(), true);
     }
 
     // enable interactive mode if reverse prompt is specified