1 files changed, 8 insertions, 6 deletions
diff --git a/src/Client/SystemCommands.h b/src/Client/SystemCommands.h
index 142ebdc..a647bed 100644
--- a/src/Client/SystemCommands.h
+++ b/src/Client/SystemCommands.h
@@ -20,15 +20,17 @@
 #ifndef MAD_CLIENT_SYSTEMCOMMANDS_H_
 #define MAD_CLIENT_SYSTEMCOMMANDS_H_
 
-#include <Common/Connection.h>
+#include <Common/XmlPacket.h>
 
-#include <string>
 #include <vector>
 
+#include <boost/shared_ptr.hpp>
 
 namespace Mad {
 namespace Client {
 
+class CommandParser;
+
 class SystemCommands {
   private:
     SystemCommands();
@@ -37,10 +39,10 @@ class SystemCommands {
     static void printHostStatus(boost::shared_ptr<const Common::XmlPacket> &packet);
 
   public:
-    static void fsinfoCommand(const std::vector<std::string> &args, Common::Connection *connection);
-    static void rebootCommand(const std::vector<std::string> &args, Common::Connection *connection);
-    static void shutdownCommand(const std::vector<std::string> &args, Common::Connection *connection);
-    static void statusCommand(const std::vector<std::string> &args, Common::Connection *connection);
+    static void fsinfoCommand(CommandParser *commandParser, const std::vector<std::string> &args);
+    static void rebootCommand(CommandParser *commandParser, const std::vector<std::string> &args);
+    static void shutdownCommand(CommandParser *commandParser, const std::vector<std::string> &args);
+    static void statusCommand(CommandParser *commandParser, const std::vector<std::string> &args);
 };
 
 }