Fix set_stack_policy test