From efcc53280f27ce0ca5d2c00b7fa96187af1607b2 Mon Sep 17 00:00:00 2001
From: champsupertramp <heychampsupertramp@gmail.com>
Date: Wed, 28 Sep 2016 20:18:02 +0800
Subject: [PATCH] Fix login redirection

---
 core/um-actions-login.php | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/core/um-actions-login.php b/core/um-actions-login.php
index be0a6422..e17816ee 100644
--- a/core/um-actions-login.php
+++ b/core/um-actions-login.php
@@ -173,8 +173,11 @@
 			exit( wp_redirect(  $args['redirect_to']  ) );
 		}
 
+		$role = um_user('role');
+		$role_data = $ultimatemember->query->role_data( $role );
+		
 		// Role redirect
-		$after = um_user('after_login');
+		$after = $role_data['after_login'];
 		switch( $after ) {
 
 			case 'redirect_admin':
@@ -186,7 +189,7 @@
 				break;
 
 			case 'redirect_url':
-				exit( wp_redirect( um_user('login_redirect_url') ) );
+				exit( wp_redirect( $role_data['login_redirect_url'] ) );
 				break;
 
 			case 'refresh':