fix bug, #1