diff --git a/examples/nlp/dialogue_state_tracking/rule_based_policy_multiwoz.py b/examples/nlp/dialogue_state_tracking/rule_based_policy_multiwoz.py
index 40181da91d5d..107775a17b76 100644
--- a/examples/nlp/dialogue_state_tracking/rule_based_policy_multiwoz.py
+++ b/examples/nlp/dialogue_state_tracking/rule_based_policy_multiwoz.py
@@ -176,9 +176,10 @@ def update_state(system_uttr, user_uttr, dial_history, belief_state):
 
 if args.mode == 'interactive':
     # for user_uttr in user_uttrs:
+    logging.info("============ Starting a new dialogue ============")
+    system_uttr, system_action, belief_state, dial_history = init_session()
     while True:
         logging.info("Type your text, use STOP to exit and RESTART to start a new dialogue.")
-        system_uttr, system_action, belief_state, dial_history = init_session()
         user_uttr = input()
 
         if user_uttr == "STOP":
diff --git a/nemo/collections/nlp/nm/non_trainables/dialogue_state_tracking/trade_state_update_nm.py b/nemo/collections/nlp/nm/non_trainables/dialogue_state_tracking/trade_state_update_nm.py
index 69700378bce9..3fc706ca6824 100755
--- a/nemo/collections/nlp/nm/non_trainables/dialogue_state_tracking/trade_state_update_nm.py
+++ b/nemo/collections/nlp/nm/non_trainables/dialogue_state_tracking/trade_state_update_nm.py
@@ -27,7 +27,7 @@
 from nemo.backends.pytorch.nm import NonTrainableNM
 from nemo.collections.nlp.data.datasets.multiwoz_dataset.multiwoz_slot_trans import REF_SYS_DA
 from nemo.collections.nlp.utils.callback_utils import tensor2numpy
-from nemo.core import AxisKind, AxisType, LogitsType, NeuralType, StringType, VoidType
+from nemo.core import AxisKind, AxisType, LogitsType, NeuralType, StringType
 from nemo.utils import logging
 from nemo.utils.decorators import add_port_docs
 
diff --git a/nemo/collections/nlp/nm/non_trainables/dialogue_state_tracking/utterance_encoder_nm.py b/nemo/collections/nlp/nm/non_trainables/dialogue_state_tracking/utterance_encoder_nm.py
index 9b4850babdfc..a9c273a17d5c 100755
--- a/nemo/collections/nlp/nm/non_trainables/dialogue_state_tracking/utterance_encoder_nm.py
+++ b/nemo/collections/nlp/nm/non_trainables/dialogue_state_tracking/utterance_encoder_nm.py
@@ -21,7 +21,7 @@
 import torch
 
 from nemo.backends.pytorch.nm import NonTrainableNM
-from nemo.core import AxisKind, AxisType, ChannelType, LengthsType, NeuralType, StringType, VoidType
+from nemo.core import AxisKind, AxisType, ChannelType, LengthsType, NeuralType, StringType
 from nemo.utils import logging
 from nemo.utils.decorators import add_port_docs