[#84] Update tests