Fix tests for rust nightly