Fix circular dependencies in sts/control_flow/__init__.py