Fix test cause implementation is ok