The normal Euclidean inverse square law is kept in a Flat universe, and we use the current (comoving) distance. However there is an additional factor of (1+z)2 reduction in brightness due to redshift and time dilation.
Figure: Whittle's (upcoming) undergraduate textbook.