minor adjustments to unit tests section