improve integration test