Adding missing 'return' keyword