minor fix
fixed integration test.