Return test batch object when saved (#229)