1 files changed, 7 insertions, 0 deletions
diff --git a/Software/Embedded_SW/Embedded/Communication/Container.c b/Software/Embedded_SW/Embedded/Communication/Container.c
index d8379fa6b..ba1f8eea9 100644
--- a/Software/Embedded_SW/Embedded/Communication/Container.c
+++ b/Software/Embedded_SW/Embedded/Communication/Container.c
@@ -39,6 +39,7 @@
 #include "Modules/Thread/Thread_ex.h"
 
 #include "Common/SWUpdate/FileSystem.h"
+#include "Common/SWUpdate/FirmwareUpgrade.h"
 #include "Communication/CommunicationTask.h"
 
 #include "StateMachines/Printing/PrintingSTM.h"
@@ -400,6 +401,12 @@ void receive_callback(char* buffer, size_t length)
          case MESSAGE_TYPE__GetFilesRequest:
              GetFilesRequestFunc(requestContainer);
              break;
+         case MESSAGE_TYPE__ActivateVersionRequest:
+             ActivateVersionRequestFunc(requestContainer);
+             break;
+         case MESSAGE_TYPE__ValidateVersionRequest:
+             ValidateVersionRequestFunc(requestContainer);
+             break;
          case MESSAGE_TYPE__StubMidTankPressureSensorRequest:
              Stub_MidTankPressureSensorRequest(requestContainer);
              break;