Use session copying in state/unit.go