Fix failing integration test
Merge pull request #331 from korczis/master

Fix failing integration test