fix for unit test. and, add more ( ) , it make me feel better