Fixed integration test by changing mock behaviour

Mocks did not did realistic stuff