Move sin(th) scaling to a later stage so direct beam does not get scaled