Adding check for type of test.