Fixing missing return statement.