[#31] fix tests