diff --git a/aes-32bit/aes.cpp b/aes-32bit/aes.cpp
index e86abc9..2b3d10f 100644
--- a/aes-32bit/aes.cpp
+++ b/aes-32bit/aes.cpp
@@ -188,7 +188,7 @@ void aes(uint8_t *in, uint8_t *out, uint8_t *skey)
 //****************************
 int main(int argc, char* argv[])
 {
-    if (argc > 1) {
+    if (argc > 2) {
         std::cerr << "Invalid number of arguments\n";
         exit(EXIT_FAILURE);
     }