In a normal form game with two players, we call a joint strategy $s=(s_1,s_2)$ a Pareto optimal outcome if for no joint strategy $s'$, for all players $i\in\{1,2\}$, we have $$u_i(s')\geq u_i(s)\qquad(1)$$ and also at least for one player $i$ $$u_i(s')>u_i(s).\qquad(2)$$
The social welfare of $s$ is defined as $$\sum_{i=1}^{2}u_i(s).\qquad(3)$$ If the social welfare of $s$ is maximal, then the joint strategy $s$ is a social optimum.
I want to use these definitions in a Bayesian game where the utility functions are defined as $u_i:A\times\Theta\to\mathbb{R}$, where $A$ is the action set and $\Theta$ is the type set.
Can we simply replace $u_i(s)$ in $(1),(2),(3)$ with $u_i(a,\theta)$? Or we have to consider the expected utility of a player?
How should we write the definitions if the game is in the ex-ante stage?