Fix return statement.