Unit test modificictaions for Unity