Update TestHelper.cs