Fix broken integration tests

Signed-off-by: Daniel Nephin <dnephin@gmail.com>