[#84] Fix bug