Fix integration test
Merge pull request #86 from leftclickben/FixIntegrationTest

Fix integration test