Bugfix

Updates #5. Re-added the missing return. This LITERALLY broke
everything. /sigh.