Update Unit Tests for TestBatch