Balancing damage and correcting state types