Fixing integration test

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