[#1] fix bug