[#8] fixed code