2 files changed, 10 insertions, 0 deletions
diff --git a/test/decrypt_block_aes.cpp b/test/decrypt_block_aes.cpp
index ff2d494..3bb8e44 100644
--- a/test/decrypt_block_aes.cpp
+++ b/test/decrypt_block_aes.cpp
@@ -109,6 +109,11 @@ int main(int argc, char** argv)
         std::cerr << "Usage error: " << e.what() << "\n";
         return 1;
     }
+    catch (const aesni::Error& e)
+    {
+        std::cerr << e;
+        return 1;
+    }
     catch (const std::exception& e)
     {
         std::cerr << e.what() << "\n";
diff --git a/test/encrypt_block_aes.cpp b/test/encrypt_block_aes.cpp
index fc56019..6de9288 100644
--- a/test/encrypt_block_aes.cpp
+++ b/test/encrypt_block_aes.cpp
@@ -109,6 +109,11 @@ int main(int argc, char** argv)
         std::cerr << "Usage error: " << e.what() << "\n";
         return 1;
     }
+    catch (const aesni::Error& e)
+    {
+        std::cerr << e;
+        return 1;
+    }
     catch (const std::exception& e)
     {
         std::cerr << e.what() << "\n";