improving integration tests