add missing return for test mode
Merge pull request #12 from joel/working

add missing return for test mode