Fixed the rust api changes