[PATCH 3/4] Fix a unit test