diff --git a/binaryai.py b/binaryai.py
index 809af45..e16eeda 100644
--- a/binaryai.py
+++ b/binaryai.py
@@ -417,7 +417,7 @@ def screen_ea_changed(self, ea, prev_ea):
             return
         func = idaapi.get_func(ea)
         if func and func.start_ea in self.plugin.function_dict:
-            if self.current_func != func:
+            if not self.current_func or self.current_func != func:
                 f = self.plugin.function_dict[func.start_ea]
                 self.plugin.viewer.update(f)
                 self.current_func = func