Fix forgotten return