[#262] Fix test