Fixes integration tests