Fix to make sure the hyperv agent and the funcitonal tests are working after the unit tests update.