fix bug, #2