Adds switch case for test.