added integration test
Tweak integration test a bit