Mihir | Putting in place a proper integration test