modify integration testing code