[#55] Update tests