opt: Optimized the unit tests.