Return fix.

Fixing the return.