[SSR] Transformation mimicking oldSetBatchSize logic (#2597)