Fix store locals in generator state