Fixing broken integration tests

Signed-off-by: Don Kjer <don.kjer@gmail.com>