Updated import to reflect new package location of TestRestFlow.