Fixed missing 'return'