Fixed last integration test which was still failing... just needed to re-structure the test