Modify to take only a single test case