diff --git a/server/src/tools/semantic_tokens.py b/server/src/tools/semantic_tokens.py
index a5a0c68..dc1a6f5 100644
--- a/server/src/tools/semantic_tokens.py
+++ b/server/src/tools/semantic_tokens.py
@@ -76,8 +76,6 @@ class _Highlighter(Visitor):
                     )
                 )
             )
-            pass
-
         if routine.contents == "set" and command.args:
             first_arg = command.args[0]
             if hasattr(first_arg, "pos") and hasattr(first_arg, "value"):
@@ -140,7 +138,6 @@ class _Highlighter(Visitor):
                                         [TokenModifier.declaration],
                                     )
                                 )
-
         if routine.contents == "namespace" and command.args:
             first_arg = command.args[1]
             if hasattr(first_arg, "pos") and hasattr(first_arg, "value"):
@@ -149,9 +146,6 @@ class _Highlighter(Visitor):
                     (((line - 1, col - 1), len(first_arg.value), "class", []))
                 )
 
-    def visit_var_sub(self, var_sub):
-        pass
-
     def tokens(self) -> list[Token]:
         """Encode tokens as described in
         https://microsoft.github.io/language-server-protocol/specifications/lsp/3.17/specification/#textDocument_semanticTokens.
diff --git a/test/test.tcl b/test/test.tcl
index 9a65346..a958218 100644
--- a/test/test.tcl
+++ b/test/test.tcl
@@ -1,4 +1,10 @@
-set main 1
+# set ::custom_flag(from_move,$::mom_path_name) 1
+# set ::custom_flag(from_move,$::mom_path_name) 1
+
+set te875st 11111
+
+#set ::custom_flag(from_move,$::mom_path_name) 1
+
 if {$main == 1 && 1 == 1} {
     puts "main"
 }
@@ -93,7 +99,7 @@ proc SERVICE_output_handling {handler} {
 #_________________________________________________________________________________________________
 proc SERVICE_get_tool_data {} {
     global mom_tool_data
-global mom_operation_info
+    global mom_operation_info
 
     set mom_tool_data(toollist) ""
     set operations $::mom_operation_name_list
@@ -103,3 +109,8 @@ global mom_operation_info
         }
     }
 }
+
+
+LIB_GE_command_buffer_edit_replace MOM_end_of_program_LIB END_OF_PROGRAM @END_OF_PROG {
+    MOM_do_template "end_of_program_rewind"
+} EndOfProgramRewind