diff --git a/src/app.hpp b/src/app.hpp
index e95253a..b066dd6 100644
--- a/src/app.hpp
+++ b/src/app.hpp
@@ -97,9 +97,4 @@ typedef struct {
 void runInferenceApp(AppCliArgs *args, void (*handler)(AppInferenceContext *context));
 void runWorkerApp(AppCliArgs *args);
 
-class BadArgumentException : public std::runtime_error {
-public:
-    explicit BadArgumentException(const std::string& message) : std::runtime_error(message) {}
-};
-
 #endif
diff --git a/src/dllama-api.cpp b/src/dllama-api.cpp
index e67ac1f..8c67430 100644
--- a/src/dllama-api.cpp
+++ b/src/dllama-api.cpp
@@ -564,9 +564,8 @@ int main(int argc, char *argv[]) {
             return EXIT_SUCCESS;
         }
         runInferenceApp(&args, server);
-    } catch (const BadArgumentException& e) {
-        fprintf(stderr, "%s\n\n", e.what());
-        usage();
+    } catch (std::exception &e) {
+        printf("🚨 Critical error: %s\n", e.what());
         cleanupSockets();
         return EXIT_FAILURE;
     }