[#94] Fix tests