Fix add_host tests