[#3] Fixed a bug