fix : add updated tests