Added forgotten return statement