revising test framework for segment