Improve how unit tests are built