fix bug, #4