Improved test_update_host

Change-Id: Ib1bbdd906dcd4ea52e0a1781afe8da5de1c14962