Fix code broken by rust update...