Fixed a small bug in policy_alpha_vectors