Added in missing return statement.