WPMU membership2 set different redirect url for certain membership

code.php

<?php
/**
  * Change $redirect_url and $membership_id variable value
  * at line 22 and 23
  */

add_action( 'login_redirect', 'custom_filter_url_after_login', 9999, 3 );
function custom_filter_url_after_login( $url, $request, $user )
{
        if( ! isset( $user->ID ) )
        {
                return $url;
        }
        
        $member = MS_Factory::load( 'MS_Model_Member', $user->ID );
        
        if( count( $member->subscriptions ) < 1 )
        {
                return $url;
        }
        
        $redirect_url = 'http://m2mu.dev/category/test/';
        $membership_id = 1784;
        
        foreach( $member->subscriptions as $subscription )
        {
                $membership = MS_Factory::load( 'MS_Model_Membership', $subscription->membership_id );
                if( $membership->id == $membership_id )
                {
                        return $redirect_url;
                }
        }
        
        return $url;
}