fix bug
fix bug 2