[DF-2998] Small fix after review.