Add missing return statement

This fixes issue #35.