fix bug 2