Fixed missing return statement.