fix bug in