[#66] Update tests