Effect of demagnetization factors on spin current transport