Fix rust setup