From f5e7d6016aff6d77b101c74a2be705d116663fc2 Mon Sep 17 00:00:00 2001
From: "R. Richard" <ni-richard@protonmail.ch>
Date: Thu, 8 Aug 2019 08:27:41 +0200
Subject: [PATCH] Fix condition in PR #62

---
 tfw/components/fsm/fsm_handler.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tfw/components/fsm/fsm_handler.py b/tfw/components/fsm/fsm_handler.py
index 61f901e..c9cb6eb 100644
--- a/tfw/components/fsm/fsm_handler.py
+++ b/tfw/components/fsm/fsm_handler.py
@@ -31,7 +31,8 @@ class FSMHandler:
             LOG.error('IGNORING MESSAGE: Invalid message received: %s', message)
 
     def handle_step(self, message):
-        return message if self.fsm.step(message['trigger']) else None
+        if self.fsm.step(message['trigger']):
+            return message
 
     def handle_update(self, message):
         # pylint: disable=no-self-use