I'd rather keep the System Under Test (SUT) separate from the testing code, which is a best practice. In this circumstance it results in code duplication, but I don't know if theres a better solution to that.