diff --git a/TaskB01/run b/TaskB01/run
index 3216646..f31edb0 100644
--- a/TaskB01/run
+++ b/TaskB01/run
@@ -50,7 +50,7 @@ class automata:
         return not set(self.state).isdisjoint(self.accepting_states)
 
     def get_node_transition(self, q, i):
-        return [] if not self.storage[q][i] else return self.storage[q][i]
+        return [] if not self.storage[q][i] else self.storage[q][i]
 
 
 auto = automata()