fix policy_learning_toolS