Fix forgotten return statement