diff --git a/examples/main/main.cpp b/examples/main/main.cpp
index e73c5356e..6131f5b46 100644
--- a/examples/main/main.cpp
+++ b/examples/main/main.cpp
@@ -361,7 +361,7 @@ int main(int argc, char ** argv) {
                 }
                 if (i > 0) {
                     // check if we've used up all the prompt but not all cached tokens
-                    if (embd.size() == i && n_session_consumed < session_tokens.size()) {
+                    if (embd.size() == i && n_session_consumed < (int) session_tokens.size()) {
                         // force revaluation of the last token to recalculate logits
                         i--;
                         n_past--;