Fix missing return statement

Fixes #5