Right, but this is already covered in the result key in the state return dict, we don't need this layer