[#7] Fix the bug.