modify unit test, I missed something