Fix tests for rust upgrade

r? @metajack