Added missing return. Fixes #23