2 files changed, 11 insertions, 6 deletions
diff --git a/signal-composer/signalcomposer.cpp b/signal-composer/signalcomposer.cpp
index 90e6446..59395f5 100644
--- a/signal-composer/signalcomposer.cpp
+++ b/signal-composer/signalcomposer.cpp
@@ -1,5 +1,5 @@
 /*
- * Copyright (C) 2018 Konsulko Group
+ * Copyright (C) 2018-2020 Konsulko Group
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -14,10 +14,13 @@
  * limitations under the License.
  */
 
+#include <QDebug>
+
 #include "message.h"
+#include "signalcomposermessage.h"
 #include "messageengine.h"
 #include "signalcomposer.h"
-#include "signalcomposermessage.h"
+
 
 SignalComposer::SignalComposer (QUrl &url, QObject * parent) :
     QObject(parent),
@@ -66,7 +69,7 @@ void SignalComposer::onDisconnected()
 
 void SignalComposer::onMessageReceived(MessageType type, Message *message)
 {
-    if (type == SignalComposerEventMessage) {
+    if (type == MessageType::SignalComposerEventMessage) {
         SignalComposerMessage *tmsg = qobject_cast<SignalComposerMessage*>(message);
 
         if (tmsg->isEvent()) {
diff --git a/signal-composer/signalcomposer.h b/signal-composer/signalcomposer.h
index 3a6ca25..13d26c6 100644
--- a/signal-composer/signalcomposer.h
+++ b/signal-composer/signalcomposer.h
@@ -1,5 +1,5 @@
 /*
- * Copyright (C) 2019 Konsulko Group
+ * Copyright (C) 2019, 2020 Konsulko Group
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -17,11 +17,13 @@
 #ifndef SIGNALCOMPOSER_H
 #define SIGNALCOMPOSER_H
 
-#include <QDebug>
 #include <QObject>
 #include <QJsonArray>
 
-#include "messageengine.h"
+class MessageEngine;
+class Message;
+
+enum class MessageType;
 
 class SignalComposer : public QObject
 {