Hotfix: Fix problems with integration tests