Added return which was accidentally removed.