I don't think the above is a problem. It's what democracy was designed for. At the moment voters vote in capitalists because that's what produces the best life quality. When AI takes over they can vote more for socialist, to each according to their needs policies.