FIxed test break because of #29