test: fix the unit tests