Fix return type

This is what I get for making cleanups after running my tests...sorry
Merge pull request #46 from timholy/master

Fix broken return type