fix bug 4