Fix integration test.

Signed-off-by: Dan Lorenc <lorenc.d@gmail.com>