Improved the way I am integration testing