Fix broken return type

This is what I get for making cleanups after running my tests...sorry