Fix tests for newest rust