[test_download] Fix import