corrected unittest to take into account the new changes of version 0.4.0