Added missing 'return'