Adapting for multi sample test case