Adding missing return

see #127