HOTFIX: Fix integration test