fix bug

finish