Checking for TestSet didn't work.