diff --git a/sbysrc/sby_core.py b/sbysrc/sby_core.py
index 0ce0970..d84c21c 100644
--- a/sbysrc/sby_core.py
+++ b/sbysrc/sby_core.py
@@ -548,6 +548,7 @@ class SbyConfig:
 
     def gen_setup_script(self):
         script = '# Autogenerated by SBY stage setup\n'
+        script += 'select *'
 
         def cutpoint(v):
             return f"cutpoint {v}"
@@ -577,8 +578,9 @@ class SbyConfig:
                 'invoke': invoke(v)
             }.get(k, '') + '\n'
 
-        # TODO: Invert selection, then `chformal -assert -remove`
-
+        script += 'select -set _setup %\n'
+        script += 'select @_setup %n\n'
+        script += 'chformal -assert -remove %\n'
         return script