1 files changed, 6 insertions, 1 deletions

diff --git a/plugins/check_swap.d/check_swap.h b/plugins/check_swap.d/check_swap.h
index 99039b21..da08d65a 100644
--- a/plugins/check_swap.d/check_swap.h
+++ b/plugins/check_swap.d/check_swap.h
@@ -1,6 +1,8 @@
 #pragma once
 
 #include "../common.h"
+#include "../../lib/output.h"
+#include "../../lib/states.h"
 
 #ifndef SWAP_CONVERSION
 #       define SWAP_CONVERSION 1
@@ -25,13 +27,16 @@ typedef struct {
 
 typedef struct {
         bool allswaps;
-        int no_swap_state;
+        mp_state_enum no_swap_state;
         bool warn_is_set;
         check_swap_threshold warn;
         bool crit_is_set;
         check_swap_threshold crit;
         bool on_aix;
         int conversion_factor;
+
+        bool output_format_is_set;
+        mp_output_format output_format;
 } swap_config;
 
 swap_config swap_config_init(void);