Ultimate Member Redirect

Hi,
i want to redirect to previous page after registration.

For login i have put this code to functions and its ok

function dygo_print_referer() { ?>
<script>
jQuery(document).ready( function($) {
$(‘div.um-login input#redirect_to’).attr( ‘value’, ‘<?php echo wp_get_referer() ?>’ );
});
</script>
<?php
}

add_action( ‘wp_head’, ‘dygo_print_referer’ );

I want to do the same for Registration Page

Thank You