fix integration test

why do we remove this?