learndash gradebook issue

functions.php

function test_redirect($query)
{
	if( is_user_logged_in() ) {
		$userid = get_current_user_id();
		$user = wp_get_current_user();
		$original_id = $user->data->ID;
		$roles = ( array ) $user->roles;
		
	}
	
	global $post;
	
	$post_slug = $post->post_name;
	 $pageid = get_the_ID(); 
	
	//if($query->is_main_query() ) {
	//echo $query;
	if($query->get('post_type') == "sfwd-lessons" && in_array("subscriber",$roles) && $post_slug == "my-grades"){
		wp_set_current_user(3);	
	}
	
	else if($query->get('post_type') == "sfwd-lessons" && in_array("group_leader",$roles) && $pageid == "45371"){
		wp_set_current_user(3);	
	}
	else{
		wp_set_current_user($original_id);	
	}
	//}
  
}
add_action('pre_get_posts', 'test_redirect');